diff --git a/examples/automation/manual_random_param_search_example.py b/examples/automation/manual_random_param_search_example.py index 0bb0cb67..f444edb9 100644 --- a/examples/automation/manual_random_param_search_example.py +++ b/examples/automation/manual_random_param_search_example.py @@ -3,7 +3,7 @@ from random import sample from trains import Task # Connecting TRAINS -task = Task.init(project_name='examples', task_name='Random Hyper-Parameter Search Example') +task = Task.init(project_name='examples', task_name='Random Hyper-Parameter Search Example', task_type=Task.TaskTypes.optimizer) # Create a hyper-parameter dictionary for the task params = dict() diff --git a/examples/automation/task_piping_example.py b/examples/automation/task_piping_example.py index a9c99cc4..96acc86e 100644 --- a/examples/automation/task_piping_example.py +++ b/examples/automation/task_piping_example.py @@ -2,7 +2,7 @@ from trains import Task from time import sleep # Initialize the Task Pipe's first Task used to start the Task Pipe -task = Task.init('examples', 'Simple Controller Task') +task = Task.init('examples', 'Simple Controller Task', task_type=Task.TaskTypes.controller) # Create a hyper-parameter dictionary for the task param = dict() diff --git a/examples/services/jupyter-service/execute_jupyter_notebook_server.py b/examples/services/jupyter-service/execute_jupyter_notebook_server.py index 5fa7092b..46625a8b 100644 --- a/examples/services/jupyter-service/execute_jupyter_notebook_server.py +++ b/examples/services/jupyter-service/execute_jupyter_notebook_server.py @@ -14,7 +14,7 @@ from trains import Task os.environ["TRAINS_DOCKER_IMAGE"] = "nvidia/cuda --network host" # initialize TRAINS -task = Task.init(project_name="examples", task_name="Remote Jupyter NoteBook") +task = Task.init(project_name="examples", task_name="Remote Jupyter NoteBook", task_type=Task.TaskTypes.service) # get rid of all the runtime TRAINS preserve = (