fix pathing

neuralmagic · May 3, 2024 · 54a2cde · 54a2cde
1 parent 4bfbee4
commit 54a2cde
Show file tree

Hide file tree

Showing 6 changed files with 6 additions and 6 deletions.
diff --git a/tests/sparseml/transformers/oneshot/oneshot_configs/tiny_stories_conf2.yaml b/tests/sparseml/transformers/oneshot/oneshot_configs/tiny_stories_conf2.yaml
@@ -3,4 +3,4 @@ test_type: "smoke"
 tokenize: False
 model: "TinyLlama/TinyLlama-1.1B-intermediate-step-1431k-3T"
 dataset: open_platypus
-recipe: "tests/sparseml/pytorch/oneshot/oneshot_configs/recipes/recipe.yaml"
+recipe: "tests/sparseml/transformers/oneshot/oneshot_configs/recipes/recipe.yaml"
diff --git a/tests/sparseml/transformers/oneshot/oneshot_configs/tiny_stories_conf3.yaml b/tests/sparseml/transformers/oneshot/oneshot_configs/tiny_stories_conf3.yaml
@@ -4,4 +4,4 @@ tokenize: False
 model: "TinyLlama/TinyLlama-1.1B-intermediate-step-1431k-3T"
 dataset: "gsm8k"
 dataset_config_name: "main"
-recipe: "tests/sparseml/pytorch/oneshot/oneshot_configs/recipes/recipe.yaml"
+recipe: "tests/sparseml/transformers/oneshot/oneshot_configs/recipes/recipe.yaml"
diff --git a/tests/sparseml/transformers/oneshot/oneshot_configs/tiny_stories_conf5.yaml b/tests/sparseml/transformers/oneshot/oneshot_configs/tiny_stories_conf5.yaml
@@ -3,4 +3,4 @@ test_type: "smoke"
 tokenize: True
 model: "TinyLlama/TinyLlama-1.1B-intermediate-step-1431k-3T"
 dataset: open_platypus
-recipe: "tests/sparseml/pytorch/oneshot/oneshot_configs/recipes/recipe.yaml"
+recipe: "tests/sparseml/transformers/oneshot/oneshot_configs/recipes/recipe.yaml"
diff --git a/tests/sparseml/transformers/oneshot/oneshot_configs/tiny_stories_conf6.yaml b/tests/sparseml/transformers/oneshot/oneshot_configs/tiny_stories_conf6.yaml
@@ -3,4 +3,4 @@ test_type: "smoke"
 tokenize: True
 model: "TinyLlama/TinyLlama-1.1B-intermediate-step-1431k-3T"
 dataset: "gsm8k"
-recipe: "tests/sparseml/pytorch/oneshot/oneshot_configs/recipes/recipe.yaml"
+recipe: "tests/sparseml/transformers/oneshot/oneshot_configs/recipes/recipe.yaml"
diff --git a/tests/sparseml/transformers/oneshot/test_api_inputs.py b/tests/sparseml/transformers/oneshot/test_api_inputs.py
@@ -22,7 +22,7 @@
 from tests.testing_utils import parse_params, requires_torch
 
 
-CONFIGS_DIRECTORY = "tests/sparseml/pytorch/oneshot/oneshot_configs"
+CONFIGS_DIRECTORY = "tests/sparseml/transformers/oneshot/oneshot_configs"
 
 # TODO: Seems better to mark test type (smoke, sanity, regression) as a marker as
 # opposed to using a field in the config file?

diff --git a/tests/sparseml/transformers/oneshot/test_cli.py b/tests/sparseml/transformers/oneshot/test_cli.py
@@ -21,7 +21,7 @@
 from tests.testing_utils import parse_params, requires_torch, run_cli_command
 
 
-CONFIGS_DIRECTORY = "tests/sparseml/pytorch/oneshot/oneshot_configs"
+CONFIGS_DIRECTORY = "tests/sparseml/transformers/oneshot/oneshot_configs"
 
 
 @pytest.mark.smoke