Enable type checking

lightly-ai · Nov 20, 2023 · d5fb239 · d5fb239
1 parent 83e610c
commit d5fb239
Showing 1 changed file with 0 additions and 7 deletions.
diff --git a/mypy.ini b/mypy.ini
@@ -52,7 +52,6 @@ exclude = (?x)(
     lightly/cli/_helpers.py |
     lightly/loss/ntx_ent_loss.py |
     lightly/loss/vicreg_loss.py |
-    lightly/loss/memory_bank.py |
     lightly/loss/tico_loss.py |
     lightly/loss/pmsn_loss.py |
     lightly/loss/swav_loss.py |
@@ -110,15 +109,9 @@ exclude = (?x)(
     lightly/utils/benchmarking/knn_classifier.py |
     lightly/utils/benchmarking/online_linear_classifier.py |
     lightly/utils/embeddings_2d.py |
-    lightly/models/_momentum.py |
-    lightly/models/batchnorm.py |
-    lightly/models/modules/heads.py |
     lightly/models/modules/masked_autoencoder.py |
-    lightly/models/modules/nn_memory_bank.py |
     lightly/models/modules/ijepa.py |
-    lightly/models/zoo.py |
     lightly/models/utils.py |
-    lightly/models/resnet.py |
     tests/cli/test_cli_version.py |
     tests/cli/test_cli_magic.py |
     tests/cli/test_cli_crop.py |