pub_soft/foldingdiff
3
0
Fork 0
mirror of https://github.com/microsoft/foldingdiff.git synced 2026-06-04 13:30:33 +08:00
Code Issues Packages Projects Releases Wiki Activity

Create plot dir if it does not yet exist

Browse Source
This commit is contained in:
Kevin Wu
2022-09-29 03:31:08 -07:00
parent 7e37d03563
commit 03caa2e003
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
foldingdiff/plotting.py
View File

@@ -19,7 +19,8 @@ import torch
from torch.utils.data import Dataset
PLOT_DIR = Path(os.path.join(os.path.dirname(os.path.dirname(__file__)), "plots"))
assert PLOT_DIR.is_dir()
if not PLOT_DIR.is_dir():
os.makedirs(PLOT_DIR)
def plot_joint_kde(