From 54a908ad6eabc69d1ee0673b0e3fabab6c71bc9a Mon Sep 17 00:00:00 2001
From: fynnbe <thefynnbe@gmail.com>
Date: Mon, 11 Nov 2024 21:31:27 +0100
Subject: [PATCH] simplify example models

---
 .../models/unet2d_diff_output_shape/environment.yaml           | 3 ---
 .../models/unet2d_diff_output_shape/v0_4.bioimageio.yaml       | 2 +-
 .../models/unet2d_fixed_shape/environment.yaml                 | 3 ---
 .../models/unet2d_fixed_shape/v0_4.bioimageio.yaml             | 2 +-
 4 files changed, 2 insertions(+), 8 deletions(-)
 delete mode 100644 example_descriptions/models/unet2d_diff_output_shape/environment.yaml
 delete mode 100644 example_descriptions/models/unet2d_fixed_shape/environment.yaml

diff --git a/example_descriptions/models/unet2d_diff_output_shape/environment.yaml b/example_descriptions/models/unet2d_diff_output_shape/environment.yaml
deleted file mode 100644
index 3bda0cbc3..000000000
--- a/example_descriptions/models/unet2d_diff_output_shape/environment.yaml
+++ /dev/null
@@ -1,3 +0,0 @@
-channels: [pytorch, conda-forge, nodefaults]
-dependencies: pytorch>=1.6,<2.0
-name: torch-em-deploy
diff --git a/example_descriptions/models/unet2d_diff_output_shape/v0_4.bioimageio.yaml b/example_descriptions/models/unet2d_diff_output_shape/v0_4.bioimageio.yaml
index 0172a73a6..3b9b3adf9 100644
--- a/example_descriptions/models/unet2d_diff_output_shape/v0_4.bioimageio.yaml
+++ b/example_descriptions/models/unet2d_diff_output_shape/v0_4.bioimageio.yaml
@@ -4,7 +4,6 @@ cite:
   - { text: training library, doi: "10.5281/zenodo.5108853" }
   - { text: architecture, doi: "10.1007/978-3-319-24574-4_28" }
 covers: [cover.jpg]
-dependencies: conda:environment.yaml
 description: Different output shape
 documentation: documentation.md
 format_version: 0.4.0
@@ -44,3 +43,4 @@ weights:
     kwargs: { depth: 3, in_channels: 1, initial_features: 16, out_channels: 1 }
     sha256: 2c475874f358eb75dc5f3b6af8d186e4fbf8da3acf43fb6662f0d5de21b0a838
     source: weights.pt
+    pytorch_version: 1.6
diff --git a/example_descriptions/models/unet2d_fixed_shape/environment.yaml b/example_descriptions/models/unet2d_fixed_shape/environment.yaml
deleted file mode 100644
index 3bda0cbc3..000000000
--- a/example_descriptions/models/unet2d_fixed_shape/environment.yaml
+++ /dev/null
@@ -1,3 +0,0 @@
-channels: [pytorch, conda-forge, nodefaults]
-dependencies: pytorch>=1.6,<2.0
-name: torch-em-deploy
diff --git a/example_descriptions/models/unet2d_fixed_shape/v0_4.bioimageio.yaml b/example_descriptions/models/unet2d_fixed_shape/v0_4.bioimageio.yaml
index 321c85228..1c569813b 100644
--- a/example_descriptions/models/unet2d_fixed_shape/v0_4.bioimageio.yaml
+++ b/example_descriptions/models/unet2d_fixed_shape/v0_4.bioimageio.yaml
@@ -41,7 +41,6 @@ weights:
   pytorch_state_dict:
     architecture: unet.py:UNet2d
     architecture_sha256: 3aeda0ace2e9a595025fa103a3eae2fd620a95924b3fba00ff5d4451acf4ad53
-    dependencies: conda:environment.yaml
     kwargs:
       {
         depth: 3,
@@ -55,3 +54,4 @@ weights:
       }
     sha256: bcf26f08c26f5ef8f891c08dba2c8812bec7991643fb7c0f7b5d3e7dc321092f
     source: weights.pt
+    pytorch_version: 1.6