Setup upper bound on numpy import in TestChangingSchemaSpark (#1120)

dbt-labs · Oct 15, 2024 · e6e1e10 · e6e1e10
1 parent 257e390
commit e6e1e10
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/functional/adapter/test_python_model.py b/tests/functional/adapter/test_python_model.py
@@ -67,7 +67,7 @@ def model(dbt, spark):
                 "ResourceClass": "SingleNode"
             }
         },
-        packages=['spacy', 'torch', 'pydantic>=1.10.8']
+        packages=['spacy', 'torch', 'pydantic>=1.10.8', 'numpy<2']
     )
     data = [[1,2]] * 10
     return spark.createDataFrame(data, schema=['test', 'test2'])