add get_fname helper function (#3659)

torch_geometric/graphgym/__init__.py

@@ -3,7 +3,8 @@
 from .utils import *  # noqa
 from .checkpoint import load_ckpt, save_ckpt, clean_ckpt
 from .cmd_args import parse_args
-from .config import cfg, set_cfg, load_cfg, dump_cfg, set_run_dir, set_agg_dir
+from .config import (cfg, set_cfg, load_cfg, dump_cfg, set_run_dir,
+                     set_agg_dir, get_fname)
 from .init import init_weights
 from .loader import create_loader
 from .logger import set_printing, create_logger
@@ -28,6 +29,7 @@
     'dump_cfg',
     'set_run_dir',
     'set_agg_dir',
+    'get_fname',
     'init_weights',
     'create_loader',
     'set_printing',

torch_geometric/graphgym/config.py

@@ -489,20 +489,31 @@ def makedirs_rm_exist(dir):
     os.makedirs(dir, exist_ok=True)
 
 
-def set_run_dir(out_dir, fname):
+def get_fname(fname):
     r"""
-    Create the directory for each random seed experiment run
+    Extract filename from file name path
 
     Args:
-        out_dir (string): Directory for output, specified in :obj:`cfg.out_dir`
         fname (string): Filename for the yaml format configuration file
-
     """
     fname = fname.split('/')[-1]
     if fname.endswith('.yaml'):
         fname = fname[:-5]
     elif fname.endswith('.yml'):
         fname = fname[:-4]
+    return fname
+
+
+def set_run_dir(out_dir, fname):
+    r"""
+    Create the directory for each random seed experiment run
+
+    Args:
+        out_dir (string): Directory for output, specified in :obj:`cfg.out_dir`
+        fname (string): Filename for the yaml format configuration file
+
+    """
+    fname = get_fname(fname)
     cfg.run_dir = os.path.join(out_dir, fname, str(cfg.seed))
     # Make output directory
     if cfg.train.auto_resume:
@@ -521,11 +532,7 @@ def set_agg_dir(out_dir, fname):
         fname (string): Filename for the yaml format configuration file
 
     """
-    fname = fname.split('/')[-1]
-    if fname.endswith('.yaml'):
-        fname = fname[:-5]
-    elif fname.endswith('.yml'):
-        fname = fname[:-4]
+    fname = get_fname(fname)
     return os.path.join(out_dir, fname)