Browse Source

[MNT] fix some details

tags/v0.3.2
bxdd 2 years ago
parent
commit
f91c247159
6 changed files with 6 additions and 6 deletions
  1. +1
    -1
      examples/dataset_image_workflow/example_files/example_yaml.yaml
  2. +1
    -1
      learnware/market/easy2/organizer.py
  3. +1
    -1
      learnware/specification/__init__.py
  4. +1
    -0
      learnware/specification/regular/__init__.py
  5. +1
    -2
      learnware/specification/regular/table/rkme.py
  6. +1
    -1
      tests/test_specification/test_rkme.py

+ 1
- 1
examples/dataset_image_workflow/example_files/example_yaml.yaml View File

@@ -2,7 +2,7 @@ model:
class_name: Model
kwargs: {}
stat_specifications:
- module_path: learnware.specification.image
- module_path: learnware.specification
class_name: RKMEImageStatSpecification
file_name: rkme.json
kwargs: {}

+ 1
- 1
learnware/market/easy2/organizer.py View File

@@ -20,7 +20,7 @@ from ... import utils
from ...config import C as conf
from ...logger import get_module_logger
from ...learnware import Learnware, get_learnware_from_dirpath
from ...specification import RKMEStatSpecification, Specification
from ...specification import Specification

from ..base import BaseOrganizer, BaseChecker
from ...logger import get_module_logger


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

@@ -1,3 +1,3 @@
from .utils import generate_stat_spec, generate_rkme_spec, generate_rkme_image_spec
from .base import Specification, BaseStatSpecification
from .regular import RKMEStatSpecification, RKMEImageStatSpecification
from .regular import RegularStatsSpecification, RKMEStatSpecification, RKMEImageStatSpecification

+ 1
- 0
learnware/specification/regular/__init__.py View File

@@ -1,2 +1,3 @@
from .table import RKMEStatSpecification
from .image import RKMEImageStatSpecification
from .base import RegularStatsSpecification

+ 1
- 2
learnware/specification/regular/table/rkme.py View File

@@ -26,8 +26,7 @@ from ....logger import get_module_logger
logger = get_module_logger("rkme")

if not _FAISS_INSTALLED:
logger.warning("Required faiss version >= 1.7.1 is not detected!")
logger.warning('Please run "conda install -c pytorch faiss-cpu" first.')
logger.warning("Required faiss version >= 1.7.1 is not detected! Please run 'conda install -c pytorch faiss-cpu' first")


class RKMEStatSpecification(RegularStatsSpecification):


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

@@ -5,13 +5,13 @@ import unittest
import tempfile
import numpy as np

import learnware
from learnware.specification import RKMEStatSpecification, RKMEImageStatSpecification
from learnware.specification import generate_rkme_image_spec, generate_rkme_spec


class TestRKME(unittest.TestCase):
def test_rkme(self):
pass
X = np.random.uniform(-10000, 10000, size=(5000, 200))
rkme = generate_rkme_spec(X)
rkme.generate_stat_spec_from_data(X)


Loading…
Cancel
Save