From 933f60b04f51f2bcf59cc2ad9a2ac70894a91800 Mon Sep 17 00:00:00 2001
From: Dominik Klein <dominik.klein@helmholtz-munich.de>
Date: Wed, 18 Jun 2025 13:43:51 +0200
Subject: [PATCH] fic import

---
 src/cellflow/preprocessing/_gene_emb.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/cellflow/preprocessing/_gene_emb.py b/src/cellflow/preprocessing/_gene_emb.py
index cbddb59f..b4aa95d4 100644
--- a/src/cellflow/preprocessing/_gene_emb.py
+++ b/src/cellflow/preprocessing/_gene_emb.py
@@ -14,7 +14,7 @@
     import torch
     from torch.utils.data import DataLoader
     from transformers import AutoTokenizer, EsmModel
-except ImportError as e:
+except (ImportError, ModuleNotFoundError) as e:
     torch = None
     DataLoader = None
     AutoTokenizer = None