Python TemplateFactory.TemplateFactory Examples

Example #1

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

Example #2

 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)

Example #3

File: cli.py Project: dsame/cli-demo

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)

Example #4

File: cli.py Project: dsame/cli-demo

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)

Example #5

File: cli.py Project: dsame/cli-demo

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)

Example #6

 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)

Example #7

File: test_dlis_config.py Project: dsame/cli-demo

    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)

Example #8

 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)

Example #9

 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)

Example #10

File: test_csharp_model.py Project: dsame/cli-demo

 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)