From 3c1a205928666cf1e84844dedd87fb241dc79fe8 Mon Sep 17 00:00:00 2001
From: Oksana Shadura <ksu.shadura@gmail.com>
Date: Wed, 11 Sep 2024 16:29:55 +0200
Subject: [PATCH] trying to enable pytorch for python 3.12

---
 coffea-dask/environment.yaml | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/coffea-dask/environment.yaml b/coffea-dask/environment.yaml
index 9fceecc..afce205 100644
--- a/coffea-dask/environment.yaml
+++ b/coffea-dask/environment.yaml
@@ -53,13 +53,15 @@ dependencies:
   - coffea=2024.8.1
   - rucio-clients
     # pyg
-  - pyg
+  #- pyg
+  - pytorch
+  - cpuonly
   - pytorch-cluster
   - pytorch-sparse
   - pytorch-spline-conv
   - pip:
     - fastjet # to be added to conda-forge: https://github.com/scikit-hep/fastjet/issues/133
     - tritonclient[all]
-    - tflite-runtime==2.14.0
+    #- tflite-runtime # not available for Python 3.12
     - onnxruntime
     - fsspec-xrootd