diff --git a/environments/tg-torch-cpu.yml b/environments/tg-torch-cpu.yml
index 1957e7d..6460612 100644
--- a/environments/tg-torch-cpu.yml
+++ b/environments/tg-torch-cpu.yml
@@ -15,7 +15,7 @@ dependencies:
   - pandas
   - pytorch
   - cpuonly
-  - pyg=2.0.4
+  - pyg
   - dgl
   - tqdm=4.64.1
   - matplotlib=3.5.3
@@ -24,7 +24,6 @@ dependencies:
   - ipycytoscape=1.3.3
   - pip:
     - pyTigerGraph
-    - tigergraph-mlworkbench
     - class-resolver==0.3.9
     - kafka-python==2.0.2
     - xgboost==1.7.1