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: