Rename classic variant to left-aligned

yutanagano · Jun 2, 2024 · d1610c8 · d1610c8
1 parent 51a81c2
commit d1610c8
Show file tree

Hide file tree

Showing 8 changed files with 11 additions and 11 deletions.
diff --git a/...r/_model_saves/SCEPTR_classic/config.json → ...del_saves/SCEPTR_left_aligned/config.json b/...r/_model_saves/SCEPTR_classic/config.json → ...del_saves/SCEPTR_left_aligned/config.json
@@ -4,7 +4,7 @@
         "initargs": {}
     },
     "model": {
-        "name": "SCEPTR (classic)",
+        "name": "SCEPTR (left-aligned)",
         "path_to_pretrained_state_dict": "model_saves/AB_SCEPTR_classic_MLM_only/state_dict.pt",
         "token_embedder": {
             "class": "CdrEmbedder",
@@ -100,4 +100,4 @@
         }
     },
     "num_epochs": 100
-}
+}
diff --git a/...ceptr/_model_saves/SCEPTR_classic/log.csv → .../_model_saves/SCEPTR_left_aligned/log.csv b/...ceptr/_model_saves/SCEPTR_classic/log.csv → .../_model_saves/SCEPTR_left_aligned/log.csv
diff --git a/..._model_saves/SCEPTR_classic/state_dict.pt → ...l_saves/SCEPTR_left_aligned/state_dict.pt b/..._model_saves/SCEPTR_classic/state_dict.pt → ...l_saves/SCEPTR_left_aligned/state_dict.pt
diff --git a/...saves/SCEPTR_classic_MLM_only/config.json → .../SCEPTR_left_aligned_MLM_only/config.json b/...saves/SCEPTR_classic_MLM_only/config.json → .../SCEPTR_left_aligned_MLM_only/config.json
@@ -4,7 +4,7 @@
         "initargs": {}
     },
     "model": {
-        "name": "SCEPTR (classic, MLM only)",
+        "name": "SCEPTR (left-aligned, MLM only)",
         "path_to_pretrained_state_dict": null,
         "token_embedder": {
             "class": "CdrEmbedder",
@@ -98,4 +98,4 @@
         }
     },
     "num_epochs": 100
-}
+}
diff --git a/...del_saves/SCEPTR_classic_MLM_only/log.csv → ...aves/SCEPTR_left_aligned_MLM_only/log.csv b/...del_saves/SCEPTR_classic_MLM_only/log.csv → ...aves/SCEPTR_left_aligned_MLM_only/log.csv
diff --git a/...ves/SCEPTR_classic_MLM_only/state_dict.pt → ...CEPTR_left_aligned_MLM_only/state_dict.pt b/...ves/SCEPTR_classic_MLM_only/state_dict.pt → ...CEPTR_left_aligned_MLM_only/state_dict.pt
diff --git a/src/sceptr/variant.py b/src/sceptr/variant.py
@@ -9,12 +9,12 @@ def mlm_only():
     return load_variant("SCEPTR_MLM_only")
 
 
-def classic():
-    return load_variant("SCEPTR_classic")
+def left_aligned():
+    return load_variant("SCEPTR_left_aligned")
 
 
-def classic_mlm_only():
-    return load_variant("SCEPTR_classic_MLM_only")
+def left_aligned_mlm_only():
+    return load_variant("SCEPTR_left_aligned_MLM_only")
 
 
 def cdr3_only():
@@ -66,4 +66,4 @@ def a_sceptr():
 
 
 def b_sceptr():
-    return load_variant("B_SCEPTR")
+    return load_variant("B_SCEPTR")
diff --git a/tests/test_variants.py b/tests/test_variants.py
@@ -16,8 +16,8 @@ def dummy_data():
     (
         variant.default(),
         variant.mlm_only(),
-        variant.classic(),
-        variant.classic_mlm_only(),
+        variant.left_aligned(),
+        variant.left_aligned_mlm_only(),
         variant.cdr3_only(),
         variant.cdr3_only_mlm_only(),
         variant.large(),