From 88c67a6a9744806e8c8927d4f8607ff399a66f80 Mon Sep 17 00:00:00 2001 From: Colin Date: Tue, 15 Oct 2024 11:00:34 -0700 Subject: [PATCH] Setup upper bound on numpy import in TestChangingSchemaSpark --- tests/functional/adapter/test_python_model.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/functional/adapter/test_python_model.py b/tests/functional/adapter/test_python_model.py index 50132b883..2ca1c2211 100644 --- 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'])