diff --git a/requirements/conda_env.yml b/requirements/conda_env.yml
index 39c03e27..5ffb13b1 100644
--- a/requirements/conda_env.yml
+++ b/requirements/conda_env.yml
@@ -22,8 +22,5 @@ dependencies:
     - conda-forge::pygrinder >=0.6.4
     - conda-forge::tsdb >=0.6.1
     - conda-forge::benchpots >=0.3
+    - conda-forge::ai4ts
     - pytorch::pytorch >=1.10.0
-
-    - pip:
-        # not published on conda-forge yet
-        - ai4ts
\ No newline at end of file