Skip to content

Commit

Permalink
Set fit() method to return None
Browse files Browse the repository at this point in the history
  • Loading branch information
DiogenesAnalytics committed Dec 21, 2023
1 parent ed22acf commit 32e9798
Showing 1 changed file with 1 addition and 4 deletions.
5 changes: 1 addition & 4 deletions src/autoencoder/model/base.py
Original file line number Diff line number Diff line change
Expand Up @@ -190,14 +190,11 @@ def compile(self, **kwargs: Any) -> None:
"""Wrapper for Keras model.compile method."""
self.model.compile(**kwargs)

def fit(self, **kwargs: Any) -> keras.callbacks.History:
def fit(self, **kwargs: Any) -> None:
"""Wrapper for the Keras model.fit method."""
# store history
self._history = self.model.fit(**kwargs)

# also return it
return self._history

def predict(self, **kwargs: Any) -> Any:
"""Wrapper for the Keras model.predict method."""
return self.model.predict(**kwargs)
Expand Down

0 comments on commit 32e9798

Please sign in to comment.