chore: minor typing fix

This commit is contained in:
Matthias
2024-10-20 15:24:45 +02:00
parent 51dd7fa58e
commit 3518a4167c

View File

@@ -39,8 +39,8 @@ class BasePyTorchClassifier(BasePyTorchModel):
def __init__(self, **kwargs):
super().__init__(**kwargs)
self.class_name_to_index = None
self.index_to_class_name = None
self.class_name_to_index = {}
self.index_to_class_name = {}
def predict(
self, unfiltered_df: DataFrame, dk: FreqaiDataKitchen, **kwargs