From 2f08fad28ef1f5b93744fe167bdb9269aca0685b Mon Sep 17 00:00:00 2001 From: Kajanan Selvanesan Date: Sun, 12 May 2024 21:38:42 +0530 Subject: [PATCH] fix: Include csv within whl --- Makefile | 7 +++++++ ibat/datasets/__init__.py | 4 ++++ ibat/datasets/_datasets/__init__.py | 0 ibat/pipeline.py | 5 ++++- setup.py | 1 + 5 files changed, 16 insertions(+), 1 deletion(-) create mode 100644 ibat/datasets/_datasets/__init__.py diff --git a/Makefile b/Makefile index c38110d..affc7b5 100644 --- a/Makefile +++ b/Makefile @@ -19,6 +19,13 @@ format: lint: ./venv/bin/flake8 ./ibat +.PHONY: build-wheel +build-wheel: + rm -rf dist + rm -rf ibat.egg-info + rm -rf build + python setup.py sdist bdist_wheel + .PHONY: clean clean: rm -rf venv diff --git a/ibat/datasets/__init__.py b/ibat/datasets/__init__.py index 975d883..f295058 100644 --- a/ibat/datasets/__init__.py +++ b/ibat/datasets/__init__.py @@ -1 +1,5 @@ from .datasets import * +from . import ( + _datasets, + datasets, +) diff --git a/ibat/datasets/_datasets/__init__.py b/ibat/datasets/_datasets/__init__.py new file mode 100644 index 0000000..e69de29 diff --git a/ibat/pipeline.py b/ibat/pipeline.py index ae1a696..c475e98 100644 --- a/ibat/pipeline.py +++ b/ibat/pipeline.py @@ -347,7 +347,10 @@ def run_dt_exp( from_time = to_time - print("\rGENERATING & EXPORTING PLOTS FOR EACH BUS STOP ENDED.", flush=True) + print( + "\rGENERATING & EXPORTING PLOTS FOR EACH BUS STOP ENDED. ", + flush=True, + ) def export_mean_dt_plot_as_image( diff --git a/setup.py b/setup.py index 1a7e0c8..680762b 100644 --- a/setup.py +++ b/setup.py @@ -46,6 +46,7 @@ classifiers=classifiers, packages=find_packages(exclude=["tests", "tests.*"]), include_package_data=True, + package_data={"": ["datasets/_datasets/*.csv"]}, install_requires=requirements, project_urls={ "Source Code": "https://github.com/aaivu/ibat",