diff --git a/.github/workflows/python-package.yml b/.github/workflows/python-package.yml index b6ae0e5..0e8dbb0 100644 --- a/.github/workflows/python-package.yml +++ b/.github/workflows/python-package.yml @@ -15,9 +15,7 @@ jobs: fail-fast: false matrix: python-version: - - "3.12" - "3.11" - - "3.10" os: - "ubuntu-latest" diff --git a/milliontrees/datasets/milliontrees_dataset.py b/milliontrees/datasets/milliontrees_dataset.py index 2f51e05..0d354f0 100644 --- a/milliontrees/datasets/milliontrees_dataset.py +++ b/milliontrees/datasets/milliontrees_dataset.py @@ -127,10 +127,7 @@ def check_init(self): # Check metadata assert len(self.metadata_array.shape) == 2 assert len(self.metadata_fields) == self.metadata_array.shape[1] - - # Check that it is not both classification and detection - assert not (self.is_classification and self.is_detection) - + # For convenience, include y in metadata_fields if y_size == 1 if self.y_size == 1: assert 'y' in self.metadata_fields diff --git a/requirements.txt b/requirements.txt index b350e3b..5ca4302 100644 --- a/requirements.txt +++ b/requirements.txt @@ -10,4 +10,5 @@ tqdm>=4.53.0 scikit-learn>=0.20.0 scipy>=1.5.4 albumentations>=0.5.2 -pytest>=6.2.1 \ No newline at end of file +pytest>=6.2.1 +bumpversion \ No newline at end of file diff --git a/tests/__init__.py b/tests/__init__.py new file mode 100644 index 0000000..e69de29