more stuff

Author: calderjo

kaggle_requirements.txt

@@ -126,6 +126,7 @@ scikit-optimize
 scikit-plot
 scikit-surprise
 seaborn
+git+https://github.com/facebookresearch/segment-anything.git
 shap
 squarify
 tensorflow-cloud

tests/test_matplotlib.py

@@ -8,10 +8,6 @@
 import numpy as np
 
 class TestMatplotlib(unittest.TestCase):
-    def test_version(self):
-        # b/308525631: newer versions of Matplotlib causes learntools to fail
-        self.assertLess(StrictVersion(matplotlib.__version__), StrictVersion("3.8.0"))
-
     def test_plot(self):
         plt.plot(np.linspace(0,1,50), np.random.rand(50))
         plt.savefig("plot1.png")

tests/test_numpy.py

@@ -7,25 +7,7 @@
 from contextlib import redirect_stdout
 
 class TestNumpy(unittest.TestCase):
-    def test_version(self):
-        # b/370860329: newer versions are not capable with current tensorflow
-        self.assertEqual(StrictVersion(np.__version__), StrictVersion("1.26.4")) 
-
     def test_array(self):
         array = np.array([1, 3])
 
         self.assertEqual((2,), array.shape)
-
-    # Numpy must be linked to the MKL. (Occasionally, a third-party package will muck up the installation
-    # and numpy will be reinstalled with an OpenBLAS backing.)
-    def test_mkl(self):
-        try:
-            from numpy.distutils.system_info import get_info
-            # This will throw an exception if the MKL is not linked correctly or return an empty dict.
-            self.assertTrue(get_info("blas_mkl"))
-        except:
-            # Fallback to check if mkl is present via show_config()
-            config_out = io.StringIO()
-            with redirect_stdout(config_out):
-                np.show_config()
-            self.assertIn("mkl_rt", config_out.getvalue())