From 36ee9e128b06b2fcfd4669d64becd680e5b3a296 Mon Sep 17 00:00:00 2001
From: aditya0by0 <aditya0by0@gmail.com>
Date: Mon, 12 May 2025 19:56:22 +0200
Subject: [PATCH] fix dirname for subclasses defined in other repo

---
 chebai/preprocessing/reader.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/chebai/preprocessing/reader.py b/chebai/preprocessing/reader.py
index 2bd6931c..f19645fc 100644
--- a/chebai/preprocessing/reader.py
+++ b/chebai/preprocessing/reader.py
@@ -1,3 +1,4 @@
+import inspect
 import os
 import sys
 from abc import ABC
@@ -39,7 +40,7 @@ def __init__(
         if collator_kwargs is None:
             collator_kwargs = dict()
         self.collator = self.COLLATOR(**collator_kwargs)
-        self.dirname = os.path.dirname(__file__)
+        self.dirname = os.path.dirname(inspect.getfile(self.__class__))
         self._token_path = token_path
 
     def _get_raw_data(self, row: Dict[str, Any]) -> Any: