fix mypy

Lightning-Universe · Jan 28, 2021 · 3399abb · 3399abb
1 parent cd0e775
commit 3399abb
Show file tree

Hide file tree

Showing 9 changed files with 4 additions and 7 deletions.
diff --git a/.github/workflows/ci-testing.yml b/.github/workflows/ci-testing.yml
@@ -89,9 +89,6 @@ jobs:
     #  run: |
     #    # tox --sitepackages
     #    python -m pip install -e .
-    #    # python flash_examples/generic_task.py
-    #    # python flash_examples/tabular_data.py
-    #    # python flash_examples/text_classification.py
     #    # python flash_examples/torchvision_classifier.py
 
     - name: Upload pytest test results

diff --git a/README.md b/README.md
@@ -283,7 +283,7 @@ print(predictions)
 ```
 To run the example:
 ```bash
-python flash_examples/finetuning/text_classification.py
+python flash_examples/finetuning/classify_text.py
 ```
 
 ### Tabular classification

diff --git a/flash_examples/generic_task.py b/flash_examples/generic_task.py
@@ -17,7 +17,7 @@
 dataset = datasets.MNIST('./data', download=True, transform=transforms.ToTensor())
 
 # 3. Split the data randomly
-train, val, test = random_split(dataset, [50000, 5000, 5000])
+train, val, test = random_split(dataset, [50000, 5000, 5000])  # type: ignore
 
 # 4. Create the model
 classifier = ClassificationTask(model, loss_fn=nn.functional.cross_entropy, optimizer=optim.Adam, learning_rate=10e-3)

diff --git a/..._examples/predict/image_classification.py → flash_examples/predict/classify_image.py b/..._examples/predict/image_classification.py → flash_examples/predict/classify_image.py
diff --git a/...xamples/predict/tabular_classification.py → flash_examples/predict/classify_tabular.py b/...xamples/predict/tabular_classification.py → flash_examples/predict/classify_tabular.py
diff --git a/...h_examples/predict/text_classification.py → flash_examples/predict/classify_text.py b/...h_examples/predict/text_classification.py → flash_examples/predict/classify_text.py
diff --git a/tests/core/__init__.py b/tests/core/__init__.py
diff --git a/tests/examples/__init__.py b/tests/examples/__init__.py
diff --git a/tests/examples/test_examples.py b/tests/examples/test_examples.py
@@ -38,8 +38,8 @@ def run_test(filepath):
     "file",
     [
         "image_classification.py",
-        "tabular_classification.py",
-        # "text_classification.py" TODO: takes too long
+        "classify_tabular.py",
+        # "classify_text.py" TODO: takes too long
     ]
 )
 def test_finetune_example(tmpdir, file):