stefanches7 commited on
Commit
f6f1751
·
1 Parent(s): 0bdbd77

add argument to extract xml cli

Browse files
Files changed (1) hide show
  1. extract_dir_structure_xml.py +1 -1
extract_dir_structure_xml.py CHANGED
@@ -2,7 +2,7 @@ import sys
2
  from pathlib import Path
3
  import xml.etree.ElementTree as ET
4
 
5
- FOLDER_LOCATION = r"C:\src\AI-assisted-Neuroimaging-harmonization"
6
 
7
  def build_xml(dir_path: Path, root_level: bool = True) -> ET.Element:
8
  """
 
2
  from pathlib import Path
3
  import xml.etree.ElementTree as ET
4
 
5
+ FOLDER_LOCATION = sys.argv[1] if len(sys.argv) > 1 else "sample_dataset"
6
 
7
  def build_xml(dir_path: Path, root_level: bool = True) -> ET.Element:
8
  """