Skip to content

Commit 7a25eb9

Browse files
authored
Update check_install.py
1 parent 0ce7c78 commit 7a25eb9

File tree

1 file changed

+3
-4
lines changed

1 file changed

+3
-4
lines changed

dlclive/check_install/check_install.py

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -42,11 +42,10 @@ def main():
4242
if not display:
4343
print('Running without displaying video')
4444

45-
# make temporary directory in $HOME
46-
# TODO: why create this temp directory in $HOME?
45+
# make temporary directory in $current
4746
print("\nCreating temporary directory...\n")
48-
tmp_dir = Path().home() / 'dlc-live-tmp'
49-
tmp_dir.mkdir(mode=0o775,exist_ok=True)
47+
tmp_dir = Path.cwd() / 'dlc-live-tmp'
48+
tmp_dir.mkdir(mode=0o775, exist_ok=True)
5049

5150
video_file = str(tmp_dir / 'dog_clip.avi')
5251
model_dir = tmp_dir / 'DLC_Dog_resnet_50_iteration-0_shuffle-0'

0 commit comments

Comments
 (0)