From 95f229a2f306a9ac3add25a22430c53419f0a60d Mon Sep 17 00:00:00 2001 From: Ilia Moiseev Date: Sun, 5 Feb 2023 19:17:14 +0300 Subject: [PATCH 1/3] Fix setup script --- setup.py | 11 +++-------- 1 file changed, 3 insertions(+), 8 deletions(-) diff --git a/setup.py b/setup.py index be16868..29feece 100644 --- a/setup.py +++ b/setup.py @@ -1,9 +1,4 @@ -import os -import sys import setuptools -sys.path.append(os.path.dirname(__file__)) - -import xaib with open("README.md", "r", encoding="utf-8") as fh: @@ -11,9 +6,9 @@ setuptools.setup( name="xai-benchmark", - version=xaib.__version__, - author=xaib.__author__, - author_email=xaib.__author_email__, + version='0.1.0', + author='Ilia Moiseev', + author_email='ilia.moiseev.5@yandex.ru', license='MIT', description="Benchmark for Explainable AI methods", long_description=long_description, From 9002bd581e8dd1ba1f261ecdd87516127a2f3e7f Mon Sep 17 00:00:00 2001 From: Ilia Moiseev Date: Sun, 12 Feb 2023 18:41:43 +0300 Subject: [PATCH 2/3] Make arrays in dataloader --- xaib/utils/utils.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/xaib/utils/utils.py b/xaib/utils/utils.py index 5d14b0c..293f016 100644 --- a/xaib/utils/utils.py +++ b/xaib/utils/utils.py @@ -53,6 +53,10 @@ def __getitem__(self, index: int) -> Dict: item = self._data[i] for ch in self._channels: batch[ch].append(item[ch]) + + for ch in batch: + batch[ch] = np.array(batch[ch]) + return batch def __iter__(self): From 06b3df7cf0f77c584abb0c8ba373d6699d752c55 Mon Sep 17 00:00:00 2001 From: Ilia Moiseev Date: Sun, 12 Feb 2023 18:42:38 +0300 Subject: [PATCH 3/3] Update ignore --- .gitignore | 1 + 1 file changed, 1 insertion(+) diff --git a/.gitignore b/.gitignore index 5db40b9..81ae884 100644 --- a/.gitignore +++ b/.gitignore @@ -136,3 +136,4 @@ dmypy.json repo/ *.pkl *_ds +.vscode/ \ No newline at end of file