From 0f973b5f8e326e3b33c746476c315b6d73a37dae Mon Sep 17 00:00:00 2001
From: Caibin Sheng <43896555+CaibinSh@users.noreply.github.com>
Date: Fri, 1 Apr 2022 17:13:09 +0200
Subject: [PATCH] Develop (#12)

* fix a typo in scAR-gpu.yml
* reorganise init.py files
---
 scAR-gpu.yml          | 2 +-
 scAR/__init__.py      | 5 ++++-
 scAR/main/__init__.py | 5 +----
 setup.py              | 2 +-
 4 files changed, 7 insertions(+), 7 deletions(-)

diff --git a/scAR-gpu.yml b/scAR-gpu.yml
index 79bc182..aa00162 100644
--- a/scAR-gpu.yml
+++ b/scAR-gpu.yml
@@ -7,7 +7,7 @@ dependencies:
 - conda-forge::python>=3.8.6
 - nvidia::cudatoolkit>=11.1
 - pytorch::pytorch>=1.10.0
-- pytorch::tcorchvision>=0.9.0
+- pytorch::torchvision>=0.9.0
 - pytorch::torchaudio>=0.8.0
 - pytorch-mutex=*=cuda
 - conda-forge::tqdm>=4.62.3
diff --git a/scAR/__init__.py b/scAR/__init__.py
index 94a49f0..2cb1272 100644
--- a/scAR/__init__.py
+++ b/scAR/__init__.py
@@ -1 +1,4 @@
-from scAR.main._scAR import model
\ No newline at end of file
+# -*- coding: utf-8 -*-
+
+from scAR.main._scAR import model
+from scAR.main import _data_generater as DataSimulator
\ No newline at end of file
diff --git a/scAR/main/__init__.py b/scAR/main/__init__.py
index a04d931..aedc925 100644
--- a/scAR/main/__init__.py
+++ b/scAR/main/__init__.py
@@ -1,5 +1,2 @@
 # -*- coding: utf-8 -*-
-
-__version__ = '0.2.0'
-
-from scAR.main import _data_generater as DataSimulator
\ No newline at end of file
+__version__ = '0.2.1'
\ No newline at end of file
diff --git a/setup.py b/setup.py
index b4a6d0a..5574fd4 100644
--- a/setup.py
+++ b/setup.py
@@ -15,7 +15,7 @@
     entry_points = {
         'console_scripts': ['scar=scAR.main.__main__:main'],},
     include_package_data=True,
-    url='https://github.com/CaibinSh/scAR',
+    url='https://github.com/Novartis/scAR',
     license='MIT',
     install_requires=[
         "torch>=1.10.0",