Fix numpy issue.

dbt-labs · Oct 26, 2024 · 7fd3c19 · 7fd3c19
1 parent d7e17f8
commit 7fd3c19
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/functional/adapter/test_python_model.py b/tests/functional/adapter/test_python_model.py
@@ -48,7 +48,7 @@ def test_incremental(self, project):
 
 
 models__simple_python_model = """
-import pandas
+import pyspark.pandas as pandas
 import torch
 import spacy
 
@@ -68,7 +68,7 @@ def model(dbt, spark):
                 "ResourceClass": "SingleNode"
             }
         },
-        packages=['spacy', 'torch', 'pydantic>=1.10.8']
+        packages=['spacy', 'torch', 'pydantic>=1.10.3', 'numpy<2']
     )
     data = [[1,2]] * 10
     return spark.createDataFrame(data, schema=['test', 'test2'])