From a90539e0e742455cbe516c49de6063cfe7e79101 Mon Sep 17 00:00:00 2001
From: Andi <andi.gerken@gmail.com>
Date: Thu, 15 Sep 2022 17:27:49 +0200
Subject: [PATCH] Removed torch from imports. Fixing issue #37
---
docs/index.md | 8 +-------
setup.py | 1 -
2 files changed, 1 insertion(+), 8 deletions(-)
diff --git a/docs/index.md b/docs/index.md
index a6bcde6..0bc2998 100644
--- a/docs/index.md
+++ b/docs/index.md
@@ -4,15 +4,9 @@ This package provides you:
- Preprocessing (orientation calculation, action calculation, raycasting, ...)
- Quick and easy evaluation of behavior
- Data, which is interchangable between labs and tools. No conversions required, since units and coordinate systems are standardized.
-- No custom data import, just `include robofish.io`
+- No custom data import, just `import robofish.io`
- This package is tested extensively
-Features coming up:
-
-- Interface for unified behavior models
-- Pytorch Datasets directly from `robofish.io` files.
-
-
<a href="https://git.imp.fu-berlin.de/bioroboticslab/robofish/io" class="myButton">Code</a>
## Installation
diff --git a/setup.py b/setup.py
index ccd90bd..6989900 100644
--- a/setup.py
+++ b/setup.py
@@ -62,7 +62,6 @@ setup(
"tqdm",
"pre-commit",
"scipy",
- "torch",
],
classifiers=[
"Development Status :: 3 - Alpha",
--
GitLab