fix tests failure caused by version incompatibility (microsoft#995)

ElSolem · Apr 15, 2023 · c9fc622 · c9fc622
1 parent c780d79
commit c9fc622
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/setup.py b/setup.py
@@ -63,7 +63,7 @@
             "statsmodels>=0.12.2",
             "psutil==5.8.0",
             "dataclasses",
-            "transformers[torch]",
+            "transformers[torch]==4.26",
             "datasets",
             "nltk",
             "rouge_score",

diff --git a/test/spark/test_0sparkml.py b/test/spark/test_0sparkml.py
@@ -21,7 +21,7 @@
             .master("local[1]")
             .config(
                 "spark.jars.packages",
-                f"com.microsoft.azure:synapseml_2.12:0.10.2,org.apache.hadoop:hadoop-azure:{pyspark.__version__},com.microsoft.azure:azure-storage:8.6.6",
+                "com.microsoft.azure:synapseml_2.12:0.10.2,org.apache.hadoop:hadoop-azure:3.3.5,com.microsoft.azure:azure-storage:8.6.6",
             )
             .config("spark.jars.repositories", "https://mmlspark.azureedge.net/maven")
             .config("spark.sql.debug.maxToStringFields", "100")