Add conda env config; fix a failing Pytest

knakamura13 · Oct 13, 2024 · aa4482f · aa4482f
1 parent e5bde0e
commit aa4482f
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 1 deletion.
diff --git a/environment.yml b/environment.yml
@@ -0,0 +1,16 @@
+name: mlrose-ky
+channels:
+  - anaconda
+  - defaults
+dependencies:
+  - python>=3.10, <3.13
+  - pip
+  - pip:
+    - matplotlib~=3.8.4
+    - networkx~=3.3
+    - numpy~=1.26.4
+    - pandas~=2.2.1
+    - pytest~=8.3.2
+    - scikit-learn~=1.5.0
+    - scipy~=1.13.1
+    - setuptools~=69.5.1
diff --git a/tests/test_neural/test_nn_base.py b/tests/test_neural/test_nn_base.py
@@ -17,7 +17,8 @@ class TestNNBase:
 
     def test_nn_base_instantiation_raises(self):
         """Test that instantiating _NNBase raises TypeError due to abstract methods."""
-        with pytest.raises(TypeError, match="Can't instantiate abstract class _NNBase with abstract methods fit, predict"):
+        with pytest.raises(TypeError, match="Can't instantiate abstract class _NNBase without an implementation for abstract methods "
+                                            "'fit', 'predict'"):
             _NNBase()
 
     def test_nn_base_abstract_methods(self):