diff --git a/README.md b/README.md index af4343e..6db7514 100644 --- a/README.md +++ b/README.md @@ -1,10 +1,10 @@ -# bs_scheduler +# bs-scheduler A Batch Size Scheduler library compatible with PyTorch DataLoaders. *** -Documentation: [API Reference](https://ancestor-mithril.github.io/bs_scheduler/). +Documentation: [API Reference](https://ancestor-mithril.github.io/bs-scheduler/). @@ -60,7 +60,7 @@ pip install bs-scheduler Or from git: ``` -pip install git+https://github.com/ancestor-mithril/bs_scheduler.git@master +pip install git+https://github.com/ancestor-mithril/bs-scheduler.git@master ``` ## Licensing diff --git a/bs_scheduler/batch_size_schedulers.py b/bs_scheduler/batch_size_schedulers.py index 60a0cc5..6f379a5 100644 --- a/bs_scheduler/batch_size_schedulers.py +++ b/bs_scheduler/batch_size_schedulers.py @@ -132,7 +132,7 @@ def __init__(self, dataloader: DataLoader, batch_size_manager: Union[BatchSizeMa check_isinstance(dataloader, DataLoader) except TypeError: print("Parameter dataloader is not a DataLoader. If you really need this feature, please open an issue at " - "https://github.com/ancestor-mithril/bs_scheduler/issues and describe your use case.") + "https://github.com/ancestor-mithril/bs-scheduler/issues and describe your use case.") raise self.dataloader: DataLoader = dataloader self.verbose: bool = verbose diff --git a/docs/README.md b/docs/README.md index ad7660d..7261f48 100644 --- a/docs/README.md +++ b/docs/README.md @@ -1 +1 @@ -::: bs_scheduler \ No newline at end of file +::: bs-scheduler \ No newline at end of file diff --git a/pyproject.toml b/pyproject.toml index a4f7f57..95ee5d9 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -1,5 +1,5 @@ [project] -name = "bs_scheduler" +name = "bs-scheduler" version = "0.4.2" requires-python = ">=3.9" description = "A PyTorch Dataloader compatible batch size scheduler library." @@ -30,11 +30,11 @@ dependencies = [ "torch>=1.1.0", ] [tool.setuptools.packages.find] -include = ["bs_scheduler"] +include = ["bs-scheduler"] [project.urls] -Repository = "https://github.com/ancestor-mithril/bs_scheduler" -Issues = "https://github.com/ancestor-mithril/bs_scheduler/issues" +Repository = "https://github.com/ancestor-mithril/bs-scheduler" +Issues = "https://github.com/ancestor-mithril/bs-scheduler/issues" [project.optional-dependencies]