Merge pull request #50 from unitaryai/updated_pl_version

Updated pytorch-lightning version
unitaryai · Apr 12, 2022 · ce4f02e · ce4f02e
2 parents 7b52bb1 + 59452f1
commit ce4f02e
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/requirements.txt b/requirements.txt
@@ -1,4 +1,4 @@
-pytorch-lightning >= 1.0.0, < 1.5.0
+pytorch-lightning > 1.5.0
 torch >= 1.10.0
 transformers != 4.18.0
 kaggle >= 1.5.8

diff --git a/setup.py b/setup.py
@@ -15,7 +15,7 @@
     author_email="[email protected]",
     url="https://github.com/unitaryai/detoxify",
     install_requires=[
-        "transformers != 4.18.0", # v4.18.0 fails to properly load the finetuned weights 
+        "transformers != 4.18.0",  # v4.18.0 fails to properly load the finetuned weights
         "torch >= 1.7.0",
         "sentencepiece >= 0.1.94",
     ],

diff --git a/tests/test_trainer.py b/tests/test_trainer.py
@@ -43,7 +43,7 @@ def get_instance(module, name, config, *args, **kwargs):
         max_epochs=1,
     )
     trainer.fit(model, data_loader, valid_data_loader)
-    results = trainer.test(test_dataloaders=valid_data_loader)
+    results = trainer.test(dataloaders=valid_data_loader)
 
     return results