Skip to content

Commit

Permalink
Changing delete_source default value to False. Also verified that unr…
Browse files Browse the repository at this point in the history
…ecognized loss does not crash Google Colab due to previous commit -- Fixed #173
  • Loading branch information
motiwari committed Jan 14, 2022
1 parent d909ace commit 057f7a4
Showing 1 changed file with 4 additions and 1 deletion.
5 changes: 4 additions & 1 deletion setup.py
Original file line number Diff line number Diff line change
Expand Up @@ -211,7 +211,7 @@ def install_ubuntu_pkgs():
) # noqa: E124


def setup_colab(delete_source=True):
def setup_colab(delete_source=False):
# If we're in Google Colab, we need to manually copy over
# the prebuilt armadillo libraries
# NOTE: This only works for Colab instances with Ubuntu 18.04 Runtimes!
Expand All @@ -225,6 +225,9 @@ def setup_colab(delete_source=True):
# TODO: Remove dangerous os.system() calls
# See https://stackoverflow.com/a/51329156
install_ubuntu_pkgs()

# TODO(@motiwari): Make this a randomly-named directory
# and set delete_source=True always
repo_location = os.path.join("/", "content", "BanditPAM")
# Note the space after the git URL to separate the source and target
os.system('git clone https://github.com/ThrunGroup/BanditPAM.git ' +
Expand Down

0 comments on commit 057f7a4

Please sign in to comment.