This repository was archived by the owner on Jun 30, 2022. It is now read-only.
File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change 11language : python
22python :
33 - " 2.7"
4+ - " 3.6"
45install :
56 - pip install -U -r requirements.txt
67 - pip install coveralls
Original file line number Diff line number Diff line change @@ -2,3 +2,4 @@ requests>=2.2.1
22marshmallow >= 2.12.1
33numpy == 1.14.2
44pandas == 0.22.0
5+ future
Original file line number Diff line number Diff line change 1010from mljar .client .dataset import DatasetClient
1111from mljar .client .experiment import ExperimentClient
1212
13- from project_based_test import ProjectBasedTest
13+ from . project_based_test import ProjectBasedTest
1414
1515class ExperimentClientTest (ProjectBasedTest ):
1616
Original file line number Diff line number Diff line change 99
1010from mljar .client .project import ProjectClient
1111from mljar .client .dataset import DatasetClient
12- from project_based_test import ProjectBasedTest
12+ from . project_based_test import ProjectBasedTest
1313from mljar .exceptions import BadValueException , IncorrectInputDataException
1414from mljar .utils import MLJAR_DEFAULT_TUNING_MODE
1515from mljar import Mljar
@@ -214,7 +214,6 @@ def test_non_wait_fit(self):
214214 self .assertTrue (pred is not None )
215215 # get MSE
216216 score = self .mse (pred , self .y )
217- print 'Score' , score
218217 self .assertTrue (score < 0.99 )
219218
220219 def test_retrive_models (self ):
Original file line number Diff line number Diff line change 1212from mljar .client .result import ResultClient
1313from mljar .exceptions import BadRequestException
1414
15- from project_based_test import ProjectBasedTest
15+ from . project_based_test import ProjectBasedTest
1616
1717class ResultClientTest (ProjectBasedTest ):
1818
Original file line number Diff line number Diff line change 44import os
55import unittest
66
7- # from .project_client_test import ProjectClientTest
8- # from .dataset_client_test import DatasetClientTest
7+ from .project_client_test import ProjectClientTest
8+ from .dataset_client_test import DatasetClientTest
99from .experiment_client_test import ExperimentClientTest
10- # from .result_client_test import ResultClientTest
11- # from .mljar_test import MljarTest
10+ from .result_client_test import ResultClientTest
11+ from .mljar_test import MljarTest
1212
1313if __name__ == '__main__' :
1414 unittest .main ()
You can’t perform that action at this time.
0 commit comments