def setUp(self): self.tmp_dirpath = tempfile.mkdtemp() factory = TemplateFactory(CSHARP_LANG) factory.generate(self.tmp_dirpath, 'tensorflow', 'Release') os.chdir(self.tmp_dirpath) os.makedirs(os.path.join('.oml', 'package'), exist_ok=True) touch(os.path.join('.oml', 'package', 'file.txt'))
def test_csharp_initialize_base(self): factory = TemplateFactory(CSHARP_LANG) factory.generate(self.tmp_dirpath, 'tensorflow', 'Release') self.assertEqual( os.path.exists(os.path.join(self.tmp_dirpath, 'oml.yml')), True) self.assertEqual( os.path.exists(os.path.join(self.tmp_dirpath, 'README.md')), True) self.assertEqual( os.path.exists(os.path.join(self.tmp_dirpath, '.gitignore')), True)
def init(ctx, name, flavor, template, language, build_mode, platform): """Create a new project from a template.""" try: ctx.tracker.track_event('command', 'init') factory = TemplateFactory(language, template) factory.generate(name, flavor, build_mode, platform) except Exception as e: ctx.tracker.track_event('exception', 'init') ctx.error_log(e)
def generate_noncomp_template(ctx): """Generate non-compliant pipeline template.""" try: ctx.tracker.track_event('command', 'generate_noncomp_template') model = Model(os.getcwd(), ctx.verbose) factory = TemplateFactory(model.language) factory.generate_noncomp_pipeline_template(model.base_path, model.model_name) except Exception as e: ctx.tracker.track_event('exception', 'generate_noncomp_template') ctx.error_log(e)
def generate_multistage_template(ctx): """Generate multistage pipelines template for an existing project.""" try: ctx.tracker.track_event('command', 'generate_multistage_template') model = Model(os.getcwd(), ctx.verbose) factory = TemplateFactory(model.language) factory.generate_multistage_pipeline_template(model.base_path, model.model_name) except Exception as e: ctx.tracker.track_event('exception', 'generate_multistage_template') ctx.error_log(e)
def test_python_initialize_base(self): factory = TemplateFactory(PYTHON_LANG) factory.generate(self.tmp_dirpath, 'tensorflow') self.assertEqual( os.path.exists(os.path.join(self.tmp_dirpath, 'oml.yml')), True) self.assertEqual( os.path.exists(os.path.join(self.tmp_dirpath, 'README.md')), True) self.assertEqual( os.path.exists(os.path.join(self.tmp_dirpath, '.gitignore')), True) self.assertEqual( self.checkDependencies( os.path.join(self.tmp_dirpath, 'env-prod.yml'), 'QASOML'), False)
def setUp(self): self.tmp_dirpath = tempfile.mkdtemp() factory = TemplateFactory(PYTHON_LANG) factory.generate(self.tmp_dirpath, 'scikit-learn') os.chdir(self.tmp_dirpath) os.makedirs(os.path.join('.oml', 'package'), exist_ok=True) touch(os.path.join('.oml', 'package', 'file.txt')) # Update oml.yml base_path = find_base_path(self.tmp_dirpath) metadata = load_model_metadata(base_path) metadata['commands'] = { 'dlis': {'deploy': {'instances': INSTANCES, 'cpu-cores': CORES, 'memory-usage-in-mb': MEMORY, 'namespace': NAMESPACE, 'timeout-in-ms': TIMEOUT}} } update_model_metadata(base_path, metadata)
def test_qas_python_initialize_base(self): factory = TemplateFactory(PYTHON_LANG) factory.generate(self.tmp_dirpath, 'pytorch', 'Release', 'qas') self.assertEqual( os.path.exists(os.path.join(self.tmp_dirpath, 'README.md')), True) self.assertEqual( os.path.exists(os.path.join(self.tmp_dirpath, '.gitignore')), True) self.assertEqual( os.path.exists(os.path.join(self.tmp_dirpath, 'oml.yml')), True) self.assertEqual( os.path.exists(os.path.join(self.tmp_dirpath, 'env-dev.yml')), True) self.assertEqual( os.path.exists(os.path.join(self.tmp_dirpath, 'env-prod.yml')), True) self.assertEqual( self.checkDependencies( os.path.join(self.tmp_dirpath, 'env-prod.yml'), 'QASOML'), True) self.assertEqual( self.checkDependencies( os.path.join(self.tmp_dirpath, 'env-prod.yml'), 'pythonnet'), True)
def setUp(self): self.tmp_dirpath = tempfile.mkdtemp() factory = TemplateFactory(PYTHON_LANG) factory.generate(self.tmp_dirpath) os.chdir(self.tmp_dirpath) self.model = Model(self.tmp_dirpath)
def setUp(self): self.tmp_dirpath = tempfile.mkdtemp() factory = TemplateFactory(CSHARP_LANG) factory.generate(self.tmp_dirpath, 'tensorflow', 'Release') self.model = Model(self.tmp_dirpath)