ci-cd

Jorgedavyd · Jun 4, 2024 · 8df8c5d · 8df8c5d
1 parent a32b155
commit 8df8c5d
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 6 deletions.
diff --git a/README.md b/README.md
@@ -86,11 +86,7 @@ data: # Dataset arguments
 from lightorch.training.cli import trainer
 
 if __name__ == '__main__':
-    trainer(
-        matmul_precision = 'medium', # default
-        deterministic = True, # default for reproducibility
-        seed = 123, # default for reproducibility
-    )
+    trainer()
 ```
 exec: `python3 -m training -c config.yaml`
 

diff --git a/tests/test_supervised.py b/tests/test_supervised.py
@@ -70,7 +70,7 @@ def test_supervised() -> None:
         datamodule=DataModule,
         valid_metrics="MSE",
         datamodule_kwargs=dict(pin_memory=False, num_workers=1, batch_size=1),
-        directions=["minimize"],
+        directions="minimize",
         precision="high",
         n_trials=10,
         trianer_kwargs=dict(fast_dev_run=True),