Browse Source

[MNT] format code by black

tags/v0.3.2
Gene 2 years ago
parent
commit
f0aec7fb13
2 changed files with 4 additions and 4 deletions
  1. +1
    -1
      learnware/specification/table/__init__.py
  2. +3
    -3
      tests/test_specification/test_rkme.py

+ 1
- 1
learnware/specification/table/__init__.py View File

@@ -1 +1 @@
from .rkme import RKMEStatSpecification
from .rkme import RKMEStatSpecification

+ 3
- 3
tests/test_specification/test_rkme.py View File

@@ -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)



Loading…
Cancel
Save