Python get_pod_command_argsの例

コード例 #1

ファイル: test_pod_cmd.py プロジェクト: saiprashanths/polyaxon

 def test_get_pod_command_args(self):
     cmd1 = MagicMock()
     cmd1.cmd = "video_prediction_train --model=DNA --num_masks=1"
     cmd2 = MagicMock()
     cmd2.cmd = "/bin/bash run.sh"
     cmd3 = MagicMock()
     cmd3.cmd = """
     python3 model.py --batch_size={{ batch_size }} \
                      --num_steps={{ num_steps }} \
                      --learning_rate={{ learning_rate }} \
                      --dropout={{ dropout }} \
                      --num_epochs={{ num_epochs }} \
                      --activation={{ activation }}
     """
     assert get_pod_command_args(cmd1) == ([
         'video_prediction_train', '--model=DNA', '--num_masks=1'
     ], [])
     assert get_pod_command_args(cmd2) == (['/bin/bash', 'run.sh'], [])
     assert get_pod_command_args(cmd3) == ([
         'python3', 'model.py', '--batch_size={{', 'batch_size', '}}',
         '--num_steps={{', 'num_steps', '}}', '--learning_rate={{',
         'learning_rate', '}}', '--dropout={{', 'dropout', '}}',
         '--num_epochs={{', 'num_epochs', '}}', '--activation={{',
         'activation', '}}'
     ], [])

コード例 #2

ファイル: test_pod_cmd.py プロジェクト: nwittstruck/polyaxon

    def test_get_pod_command_args_use_args(self):
        assert get_pod_command_args(self.cmd1) == (["/bin/bash", "-c"], [
            "video_prediction_train --model=DNA --num_masks=1"
        ])
        assert get_pod_command_args(self.cmd2) == (["/bin/bash", "-c"],
                                                   ["/bin/bash run.sh"])
        assert get_pod_command_args(self.cmd3) == (["/bin/bash", "-c"], [
            "python3 model.py "
            "--batch_size={{ batch_size }} "
            "--num_steps={{ num_steps }} "
            "--learning_rate={{ learning_rate }} "
            "--dropout={{ dropout }} "
            "--num_epochs={{ num_epochs }} "
            "--activation={{ activation }}"
        ])

        assert get_pod_command_args(self.cmd4) == (["/bin/bash", "-c"], [
            "video_prediction_train --model=DNA --num_masks=1 && "
            "python3 model.py "
            "--batch_size={{ batch_size }} "
            "--num_steps={{ num_steps }} "
            "--learning_rate={{ learning_rate }} "
            "--dropout={{ dropout }} "
            "--num_epochs={{ num_epochs }} "
            "--activation={{ activation }}"
        ])

コード例 #3

ファイル: test_pod_cmd.py プロジェクト: xuduofeng/polyaxon

    def test_get_pod_command_args_list_cmd(self):
        assert get_pod_command_args(self.list_cmd1) == (["/bin/bash", "-c"], [
            "video_prediction_train --model=DNA --num_masks=1 && "
            "video_prediction_train --model=NEW --num_masks=10"
        ])
        assert get_pod_command_args(self.list_cmd2) == (["/bin/bash", "-c"], [
            "/bin/bash run1.sh && /bin/bash run2.sh && /bin/bash run3.sh"
        ])
        assert get_pod_command_args(self.list_cmd3) == (["/bin/bash", "-c"], [
            "/bin/bash run1.sh && python3 model.py "
            "--batch_size={{ batch_size }} "
            "--num_steps={{ num_steps }} "
            "--learning_rate={{ learning_rate }} "
            "--dropout={{ dropout }} "
            "--num_epochs={{ num_epochs }} "
            "--activation={{ activation }}"
        ])

        assert get_pod_command_args(self.list_cmd4) == (["/bin/bash", "-c"], [
            "/bin/bash run1.sh && video_prediction_train --model=DNA --num_masks=1 && "
            "python3 model.py "
            "--batch_size={{ batch_size }} "
            "--num_steps={{ num_steps }} "
            "--learning_rate={{ learning_rate }} "
            "--dropout={{ dropout }} "
            "--num_epochs={{ num_epochs }} "
            "--activation={{ activation }}"
        ])

        assert get_pod_command_args(self.list_cmd5) == (["/bin/bash", "-c"], [
            "/bin/bash run1.sh && video_prediction_train --model=DNA --num_masks=1 ; "
            "python3 model.py "
            "--batch_size={{ batch_size }} "
            "--num_steps={{ num_steps }} "
            "--learning_rate={{ learning_rate }} "
            "--dropout={{ dropout }} "
            "--num_epochs={{ num_epochs }} "
            "--activation={{ activation }}"
        ])

        assert get_pod_command_args(self.list_cmd6) == (["/bin/bash", "-c"], [
            "/bin/bash run1.sh && video_prediction_train --model=DNA --num_masks=1 || "
            "python3 model.py "
            "--batch_size={{ batch_size }} "
            "--num_steps={{ num_steps }} "
            "--learning_rate={{ learning_rate }} "
            "--dropout={{ dropout }} "
            "--num_epochs={{ num_epochs }} "
            "--activation={{ activation }}"
        ])

コード例 #4

ファイル: job_spawner.py プロジェクト: ttsvetanov/polyaxon

 def get_pod_command_args(self):
     return get_pod_command_args(run_config=self.spec.run)

コード例 #5

 def get_pod_command_args(self):
     return get_pod_command_args(run_config=self.spec.run)

コード例 #6

ファイル: experiment_spawner.py プロジェクト: sher-ali/polyaxon

 def get_pod_command_args(self, task_type, task_idx):
     return get_pod_command_args(run_config=self.spec.run)

コード例 #7

ファイル: experiment_spawner.py プロジェクト: ttsvetanov/polyaxon

 def get_pod_command_args(self, task_type, task_idx):
     return get_pod_command_args(run_config=self.spec.run)