Merge pull request #9 from Oxid15/develop

Develop
Oxid15 · Feb 12, 2023 · ba011be · ba011be
2 parents d64eac5 + 06b3df7
commit ba011be
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 8 deletions.
diff --git a/.gitignore b/.gitignore
@@ -136,3 +136,4 @@ dmypy.json
 repo/
 *.pkl
 *_ds
+.vscode/
diff --git a/setup.py b/setup.py
@@ -1,19 +1,14 @@
-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:
  long_description = fh.read()
 
 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='[email protected]',
  license='MIT',
  description="Benchmark for Explainable AI methods",
  long_description=long_description,

diff --git 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):