diff --git a/learnware/specification/table/__init__.py b/learnware/specification/table/__init__.py index 8c56c8e..dc94b1e 100644 --- a/learnware/specification/table/__init__.py +++ b/learnware/specification/table/__init__.py @@ -1 +1 @@ -from .rkme import RKMEStatSpecification \ No newline at end of file +from .rkme import RKMEStatSpecification diff --git a/tests/test_specification/test_rkme.py b/tests/test_specification/test_rkme.py index 2f9614b..27a9033 100644 --- a/tests/test_specification/test_rkme.py +++ b/tests/test_specification/test_rkme.py @@ -13,15 +13,15 @@ class TestRKME(unittest.TestCase): def test_rkme(self): X = np.random.uniform(-10000, 10000, size=(5000, 200)) rkme = specification.utils.generate_rkme_spec(X) - + with tempfile.TemporaryDirectory(prefix="learnware_") as tempdir: rkme_path = os.path.join(tempdir, "rkme.json") rkme.save(rkme_path) - + with open(rkme_path, "r") as f: data = json.load(f) assert data["type"] == "RKMEStatSpecification" - + rkme2 = RKMEStatSpecification() rkme2.load(rkme_path)