changes to ensure data files are present in wheel

Signed-off-by: aishwariyachakraborty <[email protected]>
IBM · Oct 3, 2024 · 230c500 · 230c500
1 parent 3e3f632
commit 230c500
Show file tree

Hide file tree

Showing 4 changed files with 15 additions and 3 deletions.
diff --git a/transforms/code/higher_order_syntactic_profiler/python/pyproject.toml b/transforms/code/higher_order_syntactic_profiler/python/pyproject.toml
@@ -48,3 +48,9 @@ markers = ["unit: unit tests", "integration: integration tests"]
 
 [tool.coverage.run]
 include = ["src/*"]
+
+[tool.setuptools.packages.find]
+where = ["src"]
+
+[tool.setuptools.package-data]
+'src' = ['template.html']
diff --git a/transforms/code/semantic_profiler/python/pyproject.toml b/transforms/code/semantic_profiler/python/pyproject.toml
@@ -43,3 +43,9 @@ markers = ["unit: unit tests", "integration: integration tests"]
 
 [tool.coverage.run]
 include = ["src/*"]
+
+[tool.setuptools.packages.find]
+where = ["src"]
+
+[tool.setuptools.package-data]
+'src' = ['*.csv']
diff --git a/transforms/code/semantic_profiler/python/src/sp_local.py b/transforms/code/semantic_profiler/python/src/sp_local.py
@@ -19,7 +19,7 @@
 # create parameters
 input_folder = os.path.abspath(os.path.join(os.path.dirname(__file__), "..", "test-data", "input"))
 
-sp_params = {"ikb_file": "src/ikb/ikb_model.csv", "null_libs_file": "src/ikb/null_libs.csv"}
+sp_params = {"ikb_file": "ikb/ikb_model.csv", "null_libs_file": "ikb/null_libs.csv"}
 
 if __name__ == "__main__":
     # Here we show how to run outside of the runtime

diff --git a/transforms/code/semantic_profiler/python/src/sp_local_python.py b/transforms/code/semantic_profiler/python/src/sp_local_python.py
@@ -34,8 +34,8 @@
     "runtime_job_id": "job_id",
     "runtime_code_location": ParamsUtils.convert_to_ast(code_location),
     # sp params
-    "sp_ikb_file": "src/ikb/ikb_model.csv",
-    "sp_null_libs_file": "src/ikb/null_libs.csv"
+    "sp_ikb_file": "ikb/ikb_model.csv",
+    "sp_null_libs_file": "ikb/null_libs.csv"
 
 }
 if __name__ == "__main__":