From 0aee22d2c1e495295ee406b3c6e8d901cb972467 Mon Sep 17 00:00:00 2001
From: Darnell Granberry <40174000+DarnellGranberry@users.noreply.github.com>
Date: Tue, 7 Jan 2025 14:25:04 -0500
Subject: [PATCH] remove conda-forge, add fsspec(torch), limit torch ver,
 v0.3.6

---
 README.md             | 4 ++--
 conda-build/meta.yaml | 4 ++--
 requirements.txt      | 2 +-
 topaz/_version.py     | 2 +-
 4 files changed, 6 insertions(+), 6 deletions(-)

diff --git a/README.md b/README.md
index 6f030d2..aaa98ef 100644
--- a/README.md
+++ b/README.md
@@ -73,7 +73,7 @@ More information on conda environments can be found [here](https://conda.io/docs
 
 To install the precompiled Topaz package and its dependencies, including pytorch:
 ```
-conda install topaz -c tbepler -c pytorch -c conda-forge
+conda install topaz fsspec -c tbepler -c pytorch
 ```
 This installs pytorch from the official channel. To install pytorch for specific cuda versions, you will need to add the 'cudatoolkit=X.X' package. E.g. to install pytorch for CUDA 9.0:
 ```
@@ -81,7 +81,7 @@ conda install cudatoolkit=9.0 -c pytorch
 ```
 or combined into a single command:
 ```
-conda install topaz cudatoolkit=9.0 -c tbepler -c pytorch -c conda-forge
+conda install topaz fsspec cudatoolkit=9.0 -c tbepler -c pytorch
 ```
 See [here](https://pytorch.org/get-started/locally/) for additional pytorch installation instructions.
 
diff --git a/conda-build/meta.yaml b/conda-build/meta.yaml
index 8d8a1f3..e25cb18 100644
--- a/conda-build/meta.yaml
+++ b/conda-build/meta.yaml
@@ -6,7 +6,7 @@ package:
 
 source:
     git_url: https://github.com/tbepler/topaz.git
-    git_tag: v0.3.5 # Update to the latest release tag
+    git_tag: v0.3.6 # Update to the latest release tag
 
 requirements:
     build:
@@ -20,7 +20,7 @@ requirements:
         - scikit-learn>=0.19.0
         - scipy>=0.17.0
         - pillow>=6.2.0
-        - pytorch>=1.0.0
+        - pytorch>=1.0.0,<=2.3.1
         - torchvision
         - tqdm>=4.65.0
         - h5py>=3.7.0
diff --git a/requirements.txt b/requirements.txt
index baf38ab..049d62f 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,4 +1,4 @@
-torch>=1.0.0
+torch>=1.0.0,<=2.3.1
 torchvision
 tqdm>=4.65.0
 h5py>=3.7.0
diff --git a/topaz/_version.py b/topaz/_version.py
index a8d4557..d7b30e1 100644
--- a/topaz/_version.py
+++ b/topaz/_version.py
@@ -1 +1 @@
-__version__ = "0.3.5"
+__version__ = "0.3.6"