WIP fix: correct pytest Jetstream filter

huggingface · Nov 25, 2024 · a4cacf6 · a4cacf6
1 parent d18dc6e
commit a4cacf6
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/text-generation-inference/tests/helpers.py b/text-generation-inference/tests/helpers.py
@@ -12,8 +12,8 @@
 
 def skip_if_jetstream_pytorch_enabled(func):
     reason = "Skipping because Jetstream PyTorch is enabled"
-    jetstream_disabled = os.getenv("JETSTREAM_PT_DISABLE") != 1
-    return pytest.mark.skipif(jetstream_disabled, reason=reason)(func)
+    jetstream_enabled = os.getenv("JETSTREAM_PT_DISABLE") != "1"
+    return pytest.mark.skipif(jetstream_enabled, reason=reason)(func)
 
 
 def prepare_model(model_id, sequence_length):