Add tests for NNGSRunner

knakamura13 · Aug 25, 2024 · 761305b · 761305b
1 parent 91791ce
commit 761305b
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 2 deletions.
diff --git a/README.md b/README.md
@@ -137,6 +137,13 @@ BibTeX entry:
                     <br />
                     <sub><b>Kyle Nakamura</b></sub>
                 </a>
+            </td>
+            <td align="center">
+                <a href="https://github.com/gitongah">
+                    <img src="https://avatars.githubusercontent.com/u/39062444?v=4" width="100;" alt="gitongah"/>
+                    <br />
+                    <sub><b>Edwin Mbaabu</b></sub>
+                </a>
             </td>
 		</tr>
 	<tbody>

diff --git a/tests/test_runners/test_nngs_runner.py b/tests/test_runners/test_nngs_runner.py
@@ -5,8 +5,6 @@
 
 import sklearn.metrics as skmt
 
-from decorators import get_short_name
-from mlrose_ky import NNGSRunner
 
 try:
     import mlrose_ky
@@ -16,6 +14,9 @@
     sys.path.append("..")
     import mlrose_ky
 
+from mlrose_ky import NNGSRunner
+from mlrose_ky.decorators import get_short_name
+
 
 class TestNNGSRunner:
     """Tests for NNGSRunner."""