Skip to content

GitLab

Unverified Commit 100b95fe authored by QuanluZhang's avatar QuanluZhang Committed by GitHub
Browse files

Fix integration test of importing data (#4448)

parent e782fe21
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
test/nni_test/nnitest/validators.py
View file @ 100b95fe
...@@ -7,7 +7,7 @@ import subprocess ...@@ -7,7 +7,7 @@ import subprocess
import json import json
import requests import requests
from nni.experiment import Experiment from nni.experiment import Experiment
from nni.tools.nnictl.updater import load_search_space from nni.tools.nnictl.updater import load_imported_data
from utils import METRICS_URL, GET_IMPORTED_DATA_URL from utils import METRICS_URL, GET_IMPORTED_DATA_URL
...@@ -42,7 +42,7 @@ class ImportValidator(ITValidator): ...@@ -42,7 +42,7 @@ class ImportValidator(ITValidator):
assert proc.returncode == 0, \ assert proc.returncode == 0, \
'`nnictl experiment import {0} -f {1}` failed with code {2}'.format(exp_id, import_data_file_path, proc.returncode) '`nnictl experiment import {0} -f {1}` failed with code {2}'.format(exp_id, import_data_file_path, proc.returncode)
imported_data = requests.get(GET_IMPORTED_DATA_URL).json() imported_data = requests.get(GET_IMPORTED_DATA_URL).json()
origin_data = load_search_space(import_data_file_path).replace(' ', '') origin_data = load_imported_data(import_data_file_path).replace(' ', '')
assert origin_data in imported_data assert origin_data in imported_data
class MetricsValidator(ITValidator): class MetricsValidator(ITValidator):
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment