Skip to content

Commit 28583eb

Browse files
committed
Remove plot lib
1 parent 3f4c978 commit 28583eb

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

iaflow/__init__.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -408,13 +408,13 @@ def train(
408408
print(f'Training time: {time.time() - start_time}')
409409
self.clear_session()
410410

411-
self.plot_history(history, f"{run_data.get('path_model', '.')}/history.png")
411+
#self.plot_history(history, f"{run_data.get('path_model', '.')}/history.png")
412412
history_path = f'{self.models_folder}/{model_name}/{run_id}/history.pkl'
413413
with open(history_path, 'wb') as f:
414414
pkl.dump(history, f)
415415
return history
416416

417-
def save(self):
417+
def __save(self):
418418
"""
419419
Future implementation
420420
"""

setup.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,7 @@
66

77
setuptools.setup(
88
name="iaflow",
9-
version='2.1.15',
9+
version='2.2.0',
1010
author="Enmanuel Magallanes Pinargote",
1111
author_email="enmanuelmag@cardor.dev",
1212
description="This library help to create models with identifiers, checkpoints, logs and metadata automatically, in order to make the training process more efficient and traceable.",

0 commit comments

Comments
 (0)