v0.0.6

AIS-Bonn · Jan 18, 2022 · ff819a1 · ff819a1
1 parent 1aa2910
commit ff819a1
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 2 deletions.
diff --git a/tests/.gitignore b/tests/.gitignore
@@ -0,0 +1 @@
+_custom.py
diff --git a/vp_suite/__about__.py b/vp_suite/__about__.py
@@ -2,7 +2,7 @@
 
 _this_year = time.strftime("%Y")
 __name__ = "vp-suite"
-_version = 0, 0, 5
+_version = 0, 0, 6
 __version__ = ".".join(map(str, _version))
 __author__ = "Andreas Boltres"
 __author_email__ = "[email protected]"

diff --git a/vp_suite/vpsuite.py b/vp_suite/vpsuite.py
@@ -28,7 +28,7 @@ class VPSuite:
 
     DEFAULT_RUN_CONFIG = (constants.PKG_RESOURCES / 'run_config.json').resolve()
 
-    def __init__(self, device="cpu"):
+    def __init__(self, device="cuda"):
         self.device = "cuda" if device == "cuda" and torch.cuda.is_available() else "cpu"
 
         self.clear_models()