Develop (#12)

* fix a typo in scAR-gpu.yml * reorganise init.py files
Novartis · Apr 1, 2022 · 0f973b5 · 0f973b5
1 parent 61b9782
commit 0f973b5
Show file tree

Hide file tree

Showing 4 changed files with 7 additions and 7 deletions.
diff --git 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
@@ -1 +1,4 @@
-from scAR.main._scAR import model
+# -*- coding: utf-8 -*-
+
+from scAR.main._scAR import model
+from scAR.main import _data_generater as DataSimulator
diff --git 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
+__version__ = '0.2.1'
diff --git 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",