Adapted regex pattern to accommodate all_but_one as a position

curt-tigges · Oct 12, 2023 · 9ae9ec5 · 9ae9ec5
1 parent b862d10
commit 9ae9ec5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/utils/store.py b/utils/store.py
@@ -16,7 +16,7 @@
 
 DIRECTION_PATTERN = (
     r'^(kmeans|pca|das|das2d|das3d|logistic_regression|mean_diff|random_direction)_'
-    r'(?:(simple_adverb|simple_book|simple_product|simple_train|simple_res|treebank_train)_(ADJ|ADV|ALL|FEEL|NOUN|VRB)_)'
+    r'(?:(simple_adverb|simple_book|simple_product|simple_train|simple_res|treebank_train)_(ADJ|ADV|ALL|FEEL|NOUN|VRB|all_but_one)_)'
     r'layer(\d*)'
     r'\.npy$'
 )