From 7fd3c19beb3cc91dc803ad7744ddc935e508c34b Mon Sep 17 00:00:00 2001
From: VersusFacit <67295367+VersusFacit@users.noreply.github.com>
Date: Fri, 25 Oct 2024 18:38:02 -0700
Subject: [PATCH] Fix numpy issue.

---
 tests/functional/adapter/test_python_model.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/tests/functional/adapter/test_python_model.py b/tests/functional/adapter/test_python_model.py
index 43631bb73..49e8e40ad 100644
--- 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'])