Move plugin models to be alongside plugins, remove legacy Sagemaker, …

…Pytorch, notebook, and Tensorflow plugins (#736) Signed-off-by: Yee Hing Tong <[email protected]>
flyteorg · Nov 4, 2021 · ba29604 · ba29604
1 parent 673e076
commit ba29604
Show file tree

Hide file tree

Showing 46 changed files with 261 additions and 2,663 deletions.
diff --git a/.github/workflows/pythonbuild.yml b/.github/workflows/pythonbuild.yml
@@ -39,7 +39,7 @@ jobs:
         pip freeze
     - name: Test with coverage
       run: |
-        coverage run -m pytest tests/flytekit/unit tests/scripts
+        coverage run -m pytest tests/flytekit/unit
     - name: Integration Tests with coverage
       # https://github.com/actions/runner/issues/241#issuecomment-577360161
       shell: 'script -q -e -c "bash {0}"'

diff --git a/Makefile b/Makefile
@@ -48,12 +48,10 @@ spellcheck:  ## Runs a spellchecker over all code and documentation
 .PHONY: test
 test: lint ## Run tests
 	pytest tests/flytekit/unit
-	pytest tests/scripts
 
 .PHONY: unit_test
 unit_test:
 	pytest tests/flytekit/unit
-	pytest tests/scripts
 
 requirements-spark2.txt: export CUSTOM_COMPILE_COMMAND := make requirements-spark2.txt
 requirements-spark2.txt: requirements-spark2.in install-piptools

diff --git a/flytekit/common/tasks/pytorch_task.py b/flytekit/common/tasks/pytorch_task.py
diff --git a/flytekit/common/tasks/sagemaker/built_in_training_job_task.py b/flytekit/common/tasks/sagemaker/built_in_training_job_task.py
diff --git a/flytekit/common/tasks/sagemaker/custom_training_job_task.py b/flytekit/common/tasks/sagemaker/custom_training_job_task.py