test

Signed-off-by: Kevin Su <[email protected]>
flyteorg · Apr 23, 2024 · ce7b807 · ce7b807
1 parent 3162b1b
commit ce7b807
Show file tree

Hide file tree

Showing 9 changed files with 12 additions and 8 deletions.
diff --git a/examples/customizing_dependencies/requirements.in b/examples/customizing_dependencies/requirements.in
@@ -1,2 +1,3 @@
 flytekit
 scikit-learn
+pandas
diff --git a/examples/data_types_and_io/requirements.in b/examples/data_types_and_io/requirements.in
@@ -1 +1,2 @@
-pandas
+pandas
+torch
diff --git a/examples/feast_integration/requirements.in b/examples/feast_integration/requirements.in
@@ -4,5 +4,5 @@ matplotlib
 flytekitplugins-deck-standard
 scikit-learn
 numpy
-feast[aws]
 boto3
+feast[aws]==0.25.0
diff --git a/examples/kfpytorch_plugin/requirements.in b/examples/kfpytorch_plugin/requirements.in
@@ -1,4 +1,5 @@
 flytekit==1.10.3
 flytekitplugins-kfpytorch
 matplotlib
-torch
+torch
+tensorboardX
diff --git a/examples/mnist_classifier/mnist_classifier/pytorch_single_node_multi_gpu.py b/examples/mnist_classifier/mnist_classifier/pytorch_single_node_multi_gpu.py
@@ -369,21 +369,19 @@ def train_mnist(rank: int, world_size: int, hp: Hyperparameters):
 if os.getenv("SANDBOX") != "":
     mem = "100Mi"
     gpu = "0"
-    storage = "500Mi"
     ephemeral_storage = "500Mi"
 else:
     mem = "30Gi"
     gpu = str(WORLD_SIZE)
     ephemeral_storage = "500Mi"
-    storage = "20Gi"
 
 
 @task(
     retries=2,
     cache=True,
     cache_version="1.2",
-    requests=Resources(gpu=gpu, mem=mem, storage=storage, ephemeral_storage=ephemeral_storage),
-    limits=Resources(gpu=gpu, mem=mem, storage=storage, ephemeral_storage=ephemeral_storage),
+    requests=Resources(gpu=gpu, mem=mem, ephemeral_storage=ephemeral_storage),
+    limits=Resources(gpu=gpu, mem=mem, ephemeral_storage=ephemeral_storage),
 )
 def pytorch_mnist_task(hp: Hyperparameters) -> TrainingOutputs:
     print("Start MNIST training:")

diff --git a/examples/modin_plugin/requirements.in b/examples/modin_plugin/requirements.in
@@ -5,3 +5,4 @@ flytekitplugins-deck-standard
 flytekitplugins-modin
 scikit-learn
 modin
+ray
diff --git a/examples/sagemaker_inference_agent/requirements.in b/examples/sagemaker_inference_agent/requirements.in
@@ -1,4 +1,4 @@
-flytekitplugins-awssagemaker>=1.12.0b2
+flytekitplugins-awssagemaker==1.12.0b2
 xgboost
 fastapi
 uvicorn

diff --git a/examples/snowflake_agent/requirements.in b/examples/snowflake_agent/requirements.in
@@ -1,2 +1,3 @@
 flytekitplugins-snowflake==1.7.0
+flytekitplugins-envd==1.9.0
 flytekit==1.7.1b1
diff --git a/examples/snowflake_plugin/requirements.in b/examples/snowflake_plugin/requirements.in
@@ -1,2 +1,3 @@
 flytekitplugins-snowflake==1.7.0
 flytekit==1.7.1b1
+flytekitplugins-envd==1.9.0
-Original file line number
+Diff line change
@@ -1 +1,2 @@
-    pandas
+    pandas
+    torch