def init(repo_path: Text, pipelines_dir: Text = None, analytics_opt_in: bool = None): """Initialize ZenML on given path.""" if repo_path is None: repo_path = os.getcwd() if analytics_opt_in is None: analytics_opt_in = confirmation( "ZenML collects anonymized usage information. This data helps us " "create a better product and understand the needs of the " "community better. You can find more information about exactly " "why, what and how we collect usage analytics statistics at: " "https://docs.zenml.io/misc/usage-analytics. " "Would you like to opt-in to usage analytics?") try: Repository.init_repo( repo_path, None, None, pipelines_dir, analytics_opt_in, ) click.echo(f'ZenML repo initialized at {repo_path}') except git.InvalidGitRepositoryError: click.echo(f'{repo_path} is not a valid git repository! Please ' f'initialize ZenML within a git repository.')
from zenml.core.steps.preprocesser.standard_preprocesser \ .standard_preprocesser import \ StandardPreprocesser from zenml.core.steps.split.categorical_domain_split_step import \ CategoricalDomainSplit from zenml.core.steps.trainer.tensorflow_trainers.tf_ff_trainer import \ FeedForwardTrainer from zenml.utils import path_utils from zenml.utils.logger import get_logger logger = get_logger(__name__) # reset pipeline root to redirect to tests so that it writes the yamls there ZENML_ROOT = str(Path(zenml.__path__[0]).parent) TEST_ROOT = os.path.join(ZENML_ROOT, "tests") Repository.init_repo(TEST_ROOT, analytics_opt_in=False) pipeline_root = os.path.join(TEST_ROOT, "pipelines") csv_root = os.path.join(TEST_ROOT, "test_data") image_root = os.path.join(csv_root, "images") repo: Repository = Repository.get_instance() if path_utils.is_dir(pipeline_root): path_utils.rm_dir(pipeline_root) repo.zenml_config.set_pipelines_dir(pipeline_root) try: for i in range(1, 6): training_pipeline = TrainingPipeline(name='csvtest{0}'.format(i)) try: