diff --git a/.domino/compiled_metadata.json b/.domino/compiled_metadata.json
index 5528966..c0f947c 100644
--- 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
index 4ee5dac..8835c33 100644
--- 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"
     }
 }
\ No newline at end of file
diff --git a/config.toml b/config.toml
index e5be3a6..dbf906a 100644
--- 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"
\ No newline at end of file
+VERSION = "0.2.1"
diff --git a/pieces/OpenCVFilterPiece/test_opencv_piece.py b/pieces/OpenCVFilterPiece/test_opencv_piece.py
index d81836b..2b387c2 100644
--- 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
index 6d23114..4f5b78e 100644
--- 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
index f0315ef..6672503 100644
--- 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
index c03e563..1a9ca34 100644
--- 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
index 1682b91..5fcca97 100644
--- 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
index ccf9008..50f3077 100644
--- 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
index 24f8c89..f564bf0 100644
--- 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
index 9f57c75..ed39c41 100644
--- 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
index e851602..5a3707e 100644
--- 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": {