permanent pip install in drive

korakot · Oct 11, 2020 · 96a9aa4 · 96a9aa4
1 parent 46f0ec7
commit 96a9aa4
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 2 deletions.
diff --git a/kora/drive.py b/kora/drive.py
@@ -115,7 +115,6 @@ def link_nbs():
     import import_ipynb
     import_ipynb.find_notebook = find_in_nbs
 
-
 def find_in_nbs(fullname, path=None):
     """ lookup a notebook file to import """
     nb_path = '/nbs/' + fullname + '.ipynb'
@@ -125,3 +124,11 @@ def find_in_nbs(fullname, path=None):
     nb_path = nb_path.replace("_", " ")
     if Path(nb_path).is_file():
         return nb_path
+
+
+def install(package):
+    """ Permanently install a package in Colab """
+    link_nbs()
+    cmd = 'pip install --target=/nbs ' + package
+    print('!'+cmd)
+    get_ipython().system(cmd)
diff --git a/setup.py b/setup.py
@@ -11,7 +11,7 @@
 
 setuptools.setup(
     name="kora", 
-    version="0.7.5",
+    version="0.7.6",
     author="Korakot Chaovavanich",
     author_email="[email protected]",
     description="Convenient tools for Colab",