Skip to content

Commit

Permalink
Update setup.py
Browse files Browse the repository at this point in the history
  • Loading branch information
geometrikal committed Feb 7, 2024
1 parent 01326d4 commit 8bbfde5
Show file tree
Hide file tree
Showing 2 changed files with 6 additions and 7 deletions.
6 changes: 3 additions & 3 deletions miso/cli/__main__.py
Original file line number Diff line number Diff line change
Expand Up @@ -4,12 +4,12 @@


@click.command()
@click.option('--model', '-m', type=click.Path(exists=True), required=True, help='Path to the model information.')
@click.option('--model', '-m', type=click.Path(exists=True), required=True, help='Path to the model information XML.')
@click.option('--input', '-i', type=click.Path(exists=True), required=True, help='Path to the directory containing images.')
@click.option('--output', '-o', type=click.Path(), required=True, help='Path where the output CSV will be saved.')
@click.option('--output', '-o', type=click.Path(), required=True, help='Path of the output CSV to be saved.')
@click.option('--batch_size', '-b', type=int, default=32, show_default=True, help='Batch size for processing images.')
@click.option('--in_samples/--no-in_samples', '-f', default=False, show_default=True, help='Set this flag if images are stored in subfolders, using the subfolder names as sample labels.')
@click.option('--sample', '-s', type=str, default='unknown', show_default=True, help='Default sample name if not using subdirectories.')
@click.option('--sample', '-s', type=str, default='unknown', show_default=True, help='Default sample name if not using subfolders.')
@click.option('--unsure_threshold', 'u', type=float, default=0.0, show_default=True, help='Threshold below which predictions are considered unsure.')
def classify_folder(model, folder, output, batch_size, in_samples, sample, unsure_threshold):
"""
Expand Down
7 changes: 3 additions & 4 deletions setup.py
Original file line number Diff line number Diff line change
@@ -1,12 +1,12 @@
from setuptools import setup
from setuptools import setup, find_packages
import pathlib

here = pathlib.Path(__file__).parent.resolve()
long_description = (here / 'README.md').read_text(encoding='utf-8')

setup(
name='miso',
version='3.1.4',
version='3.1.5',
description='Python scripts for training CNNs for particle classification',
long_description=long_description,
long_description_content_type='text/markdown',
Expand All @@ -18,8 +18,7 @@
],
keywords='microfossil, cnn',
python_requires='>=3.9,<=3.11',
packages=['miso', 'miso.data', 'miso.deploy', 'miso.layers', 'miso.models', 'miso.stats', 'miso.training',
'miso.utils'],
packages=find_packages(where='miso'),
install_requires=['tensorflow==2.10.1',
'image-classifiers==1.0.0',
'lxml==5.1.0',
Expand Down

0 comments on commit 8bbfde5

Please sign in to comment.