Merge pull request #6 from Tauffer-Consulting/dev

Dev
Tauffer-Consulting · Dec 1, 2023 · e4ff71d · e4ff71d
2 parents 212cf44 + 68be450
commit e4ff71d
Show file tree

Hide file tree

Showing 12 changed files with 41 additions and 14 deletions.
diff --git a/.domino/compiled_metadata.json b/.domino/compiled_metadata.json
@@ -457,6 +457,16 @@
             }
         },
         "description": "Applies t-SNE to the data",
+        "container_resources": {
+            "requests": {
+                "cpu": 1000,
+                "memory": 1024
+            },
+            "limits": {
+                "cpu": 1000,
+                "memory": 1024
+            }
+        },
         "input_schema": {
             "properties": {
                 "data_path": {

diff --git a/.domino/dependencies_map.json b/.domino/dependencies_map.json
@@ -15,7 +15,7 @@
             "InferenceModelPiece"
         ],
         "secrets": [],
-        "source_image": "ghcr.io/tauffer-consulting/ml-domino-pieces:0.2.0-group0"
+        "source_image": "ghcr.io/tauffer-consulting/ml-domino-pieces:development-group0"
     },
     "group1": {
         "dependency": {
@@ -26,6 +26,6 @@
             "OpenCVFilterPiece"
         ],
         "secrets": [],
-        "source_image": "ghcr.io/tauffer-consulting/ml-domino-pieces:0.2.0-group1"
+        "source_image": "ghcr.io/tauffer-consulting/ml-domino-pieces:development-group1"
     }
 }
diff --git a/config.toml b/config.toml
@@ -9,4 +9,4 @@ REPOSITORY_LABEL = "ML Pieces Repository"
 
 # The version of this Pieces release
 # Attention: changing this will create a new release
-VERSION = "0.2.0"
+VERSION = "0.2.1"
diff --git a/pieces/OpenCVFilterPiece/test_opencv_piece.py b/pieces/OpenCVFilterPiece/test_opencv_piece.py
@@ -1,4 +1,5 @@
-from domino.testing import piece_dry_run, skip_envs
+from domino.testing import piece_dry_run
+from domino.testing.utils import skip_envs
 import pandas as pd
 
 

diff --git a/pieces/PCAInferencePiece/test_pca_inference_piece.py b/pieces/PCAInferencePiece/test_pca_inference_piece.py
@@ -1,4 +1,5 @@
-from domino.testing import piece_dry_run, skip_envs
+from domino.testing import piece_dry_run
+from domino.testing.utils import skip_envs
 import pandas as pd
 
 

diff --git a/pieces/PCATrainPiece/test_pca_train_piece.py b/pieces/PCATrainPiece/test_pca_train_piece.py
@@ -1,4 +1,5 @@
-from domino.testing import piece_dry_run, skip_envs
+from domino.testing import piece_dry_run
+from domino.testing.utils import skip_envs
 import pandas as pd
 
 @skip_envs("github")

diff --git a/pieces/StandardScalerPiece/test_standard_scaler_piece.py b/pieces/StandardScalerPiece/test_standard_scaler_piece.py
@@ -1,4 +1,5 @@
-from domino.testing import piece_dry_run, skip_envs
+from domino.testing import piece_dry_run
+from domino.testing.utils import skip_envs
 import pandas as pd
 
 

diff --git a/pieces/TSNEPiece/test_tsne_piece.py b/pieces/TSNEPiece/test_tsne_piece.py
@@ -1,4 +1,5 @@
-from domino.testing import piece_dry_run, skip_envs
+from domino.testing import piece_dry_run
+from domino.testing.utils import skip_envs
 import pandas as pd
 
 @skip_envs("github")

diff --git a/pieces/TrainTestSplitPiece/test_train_test_split_piece.py b/pieces/TrainTestSplitPiece/test_train_test_split_piece.py
@@ -1,4 +1,5 @@
-from domino.testing import piece_dry_run, skip_envs
+from domino.testing import piece_dry_run
+from domino.testing.utils import skip_envs
 import pandas as pd
 
 @skip_envs("github")

diff --git a/pieces/UMAPPiece/metadata.json b/pieces/UMAPPiece/metadata.json
@@ -9,6 +9,16 @@
         "dimensionality",
         "reduction"
     ],
+    "container_resources": {
+        "requests": {
+            "cpu": 1000,
+            "memory": 1024
+        },
+        "limits": {
+            "cpu": 1000,
+            "memory": 1024
+        }
+    },
     "style": {
         "node_label": "UMAP Piece",
         "node_style": {

diff --git a/pieces/UMAPPiece/test_umap_piece.py b/pieces/UMAPPiece/test_umap_piece.py
@@ -1,4 +1,5 @@
-from domino.testing import piece_dry_run, skip_envs
+from domino.testing import piece_dry_run
+from domino.testing.utils import skip_envs
 import pandas as pd
 
 

diff --git a/pieces_to_skip_in_dev/TextSummarizerPiece/metadata.json b/pieces_to_skip_in_dev/TextSummarizerPiece/metadata.json
@@ -7,12 +7,12 @@
     "container_resources": {
         "use_gpu": true,
         "requests": {
-            "cpu": "1000m",
-            "memory": "3Gi"
+            "cpu": 1000,
+            "memory": 3072
         },
         "limits": {
-            "cpu": "5000m",
-            "memory": "15Gi"
+            "cpu": 5000,
+            "memory": 15360
         }
     },
     "style": {