diff --git a/prepare_package.py b/prepare_package.py index 377e69b68..ff1b47e0c 100644 --- a/prepare_package.py +++ b/prepare_package.py @@ -21,6 +21,11 @@ shutil.copytree( os.path.join(".", pgk_data_src), os.path.join(pkg_data_folder, pkg_data_dest) ) +# Move the MVS parameters from the docs into the package_data folder +shutil.copyfile( + os.path.join(".", "docs", "MVS_parameters_list.csv"), + os.path.join(pkg_data_folder, "MVS_parameters_list.csv"), +) # Rebuild the package os.system("python setup.py sdist bdist_wheel") diff --git a/src/multi_vector_simulator/utils/__init__.py b/src/multi_vector_simulator/utils/__init__.py index e49fc1794..12fd99095 100644 --- a/src/multi_vector_simulator/utils/__init__.py +++ b/src/multi_vector_simulator/utils/__init__.py @@ -124,12 +124,24 @@ def get_doc_type(self, param_label): try: - REPO_PATH = os.path.dirname( - os.path.dirname(os.path.dirname(os.path.dirname(os.path.abspath(__file__)))) - ) - PARAMETERS_DOC = ParameterDocumentation( - param_info_file=os.path.join(REPO_PATH, "docs", "MVS_parameters_list.csv") + mvs_parameter_file = "MVS_parameters_list.csv" + DOC_PATH = os.path.join( + os.path.dirname( + os.path.dirname(os.path.dirname(os.path.dirname(os.path.abspath(__file__)))) + ), + "docs", ) + if os.path.exists(DOC_PATH): + PARAMETERS_DOC = ParameterDocumentation( + param_info_file=os.path.join(DOC_PATH, mvs_parameter_file) + ) + elif os.path.exists(PACKAGE_DATA_PATH): + PARAMETERS_DOC = ParameterDocumentation( + param_info_file=os.path.join(PACKAGE_DATA_PATH, mvs_parameter_file) + ) + else: + PARAMETERS_DOC = None + except FileNotFoundError: PARAMETERS_DOC = None