some broken dependencies causing test to fail, trying to fix pyprojec…

…t.toml
The-Strategy-Unit · Aug 9, 2023 · 301c8b7 · 301c8b7
1 parent f1256ed
commit 301c8b7
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/poetry.lock b/poetry.lock
diff --git a/pyproject.toml b/pyproject.toml
@@ -17,7 +17,7 @@ matplotlib = "^3.3.2"
 numpy = ">=1.22"
 pandas = "^1.4.0"
 scikit-learn = "1.0.2"
-tensorflow = "^2.11.0"
+tensorflow = "2.12.0"
 transformers = "^4.26.1"
 scipy = "^1.10.1"
 xgboost = "^1.7.5"

diff --git a/tests/test_factory_pipeline.py b/tests/test_factory_pipeline.py
@@ -2,7 +2,7 @@
 
 import numpy as np
 import pytest
-from keras.src.engine.functional import Functional
+from keras.engine.functional import Functional
 from sklearn.base import is_classifier
 from sklearn.pipeline import Pipeline