├── .devcontainer ├── devcontainer.json └── docker-compose.yml ├── .github └── workflows │ └── main.yml ├── .gitignore ├── .pdbrc ├── CONTRIBUTING.md ├── LICENSE ├── README.md ├── airflow_valohai_plugin ├── __init__.py ├── hooks │ ├── __init__.py │ └── valohai_hook.py ├── operators │ ├── __init__.py │ ├── valohai_download_execution_outputs_operator.py │ └── valohai_submit_execution_operator.py └── valohai_plugin.py ├── examples ├── dags │ ├── example_valohai_dag.py │ └── example_valohai_dag_pass_outputs.py └── webserver_config.py ├── requirements-dev.txt ├── run_unit_tests.sh ├── setup.cfg ├── setup.py └── tests ├── __init__.py ├── hooks ├── __init__.py └── test_valohai_hook.py └── operators ├── __init__.py └── test_valohai_submit_execution_operator.py /.devcontainer/devcontainer.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/skillupco/airflow-valohai-plugin/HEAD/.devcontainer/devcontainer.json -------------------------------------------------------------------------------- /.devcontainer/docker-compose.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/skillupco/airflow-valohai-plugin/HEAD/.devcontainer/docker-compose.yml -------------------------------------------------------------------------------- /.github/workflows/main.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/skillupco/airflow-valohai-plugin/HEAD/.github/workflows/main.yml -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/skillupco/airflow-valohai-plugin/HEAD/.gitignore -------------------------------------------------------------------------------- /.pdbrc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/skillupco/airflow-valohai-plugin/HEAD/.pdbrc -------------------------------------------------------------------------------- /CONTRIBUTING.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/skillupco/airflow-valohai-plugin/HEAD/CONTRIBUTING.md -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/skillupco/airflow-valohai-plugin/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/skillupco/airflow-valohai-plugin/HEAD/README.md -------------------------------------------------------------------------------- /airflow_valohai_plugin/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /airflow_valohai_plugin/hooks/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /airflow_valohai_plugin/hooks/valohai_hook.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/skillupco/airflow-valohai-plugin/HEAD/airflow_valohai_plugin/hooks/valohai_hook.py -------------------------------------------------------------------------------- /airflow_valohai_plugin/operators/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /airflow_valohai_plugin/operators/valohai_download_execution_outputs_operator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/skillupco/airflow-valohai-plugin/HEAD/airflow_valohai_plugin/operators/valohai_download_execution_outputs_operator.py -------------------------------------------------------------------------------- /airflow_valohai_plugin/operators/valohai_submit_execution_operator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/skillupco/airflow-valohai-plugin/HEAD/airflow_valohai_plugin/operators/valohai_submit_execution_operator.py -------------------------------------------------------------------------------- /airflow_valohai_plugin/valohai_plugin.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/skillupco/airflow-valohai-plugin/HEAD/airflow_valohai_plugin/valohai_plugin.py -------------------------------------------------------------------------------- /examples/dags/example_valohai_dag.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/skillupco/airflow-valohai-plugin/HEAD/examples/dags/example_valohai_dag.py -------------------------------------------------------------------------------- /examples/dags/example_valohai_dag_pass_outputs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/skillupco/airflow-valohai-plugin/HEAD/examples/dags/example_valohai_dag_pass_outputs.py -------------------------------------------------------------------------------- /examples/webserver_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/skillupco/airflow-valohai-plugin/HEAD/examples/webserver_config.py -------------------------------------------------------------------------------- /requirements-dev.txt: -------------------------------------------------------------------------------- 1 | apache-airflow==2.2.4 2 | flake8==4.0.1 3 | nose==1.3.7 -------------------------------------------------------------------------------- /run_unit_tests.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/skillupco/airflow-valohai-plugin/HEAD/run_unit_tests.sh -------------------------------------------------------------------------------- /setup.cfg: -------------------------------------------------------------------------------- 1 | [flake8] 2 | max_line_length = 128 3 | ignore = E731 -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/skillupco/airflow-valohai-plugin/HEAD/setup.py -------------------------------------------------------------------------------- /tests/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/hooks/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/hooks/test_valohai_hook.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/skillupco/airflow-valohai-plugin/HEAD/tests/hooks/test_valohai_hook.py -------------------------------------------------------------------------------- /tests/operators/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/operators/test_valohai_submit_execution_operator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/skillupco/airflow-valohai-plugin/HEAD/tests/operators/test_valohai_submit_execution_operator.py --------------------------------------------------------------------------------