From d15efd7b821e7115a131f2550a2dc388fed7a7f9 Mon Sep 17 00:00:00 2001
From: Andi Gerken <andi.gerken@gmail.com>
Date: Fri, 16 Jun 2023 15:47:42 +0200
Subject: [PATCH] Fixed import issue
---
src/robofish/evaluate/evaluate.py | 5 ++---
1 file changed, 2 insertions(+), 3 deletions(-)
diff --git a/src/robofish/evaluate/evaluate.py b/src/robofish/evaluate/evaluate.py
index 326db2f..1d5b3f4 100644
--- a/src/robofish/evaluate/evaluate.py
+++ b/src/robofish/evaluate/evaluate.py
@@ -657,9 +657,8 @@ def evaluate_social_vector(
try:
from fish_models.models.pascals_lstms.attribution import SocialVectors
except ImportError:
- raise ImportError(
- "Please install the fish_models package to use this function."
- )
+ warnings.warn("Please install the fish_models package to use this function.")
+ return
if poses_from_paths is None:
poses_from_paths, file_settings = utils.get_all_poses_from_paths(
--
GitLab