careamics 0.0.5__tar.gz → 0.0.7__tar.gz

{careamics-0.0.5 → careamics-0.0.7}/.github/dependabot.yml

@@ -6,7 +6,6 @@ updates:
     schedule:
       interval: "daily"
       time: "08:00"
-    groups:
-      python-packages:
-        patterns:
-          - "*"
+    ignore:
+      - dependency-name: "numpy"
+      - dependency-name: "zarr"

careamics-0.0.7/.github/pull_request_template.md

@@ -0,0 +1,86 @@
+## Description
+
+<!-- This section provides the necessary background and information for reviewers to
+understand the code and have the correct mindset when examining changes. -->
+
+> [!NOTE]  
+> **tldr**: <!-- Write a one sentence summary. -->
+
+
+### Background - why do we need this PR?
+
+<!-- What problem are you solving? Describe in a few sentences the state before
+this PR. Use code examples if useful. -->
+
+### Overview - what changed?
+
+<!-- What aspects and mechanisms of the code base changed? Describe only the general 
+idea and overarching features. -->
+
+### Implementation - how did you implement the changes?
+
+<!-- How did you solve the issue technically? Explain why you chose this approach and 
+provide code examples if applicable (e.g. change in the API for users). -->
+
+
+## Changes Made
+
+<!-- This section highlights the important features and files that reviewers should
+pay attention to when reviewing. Only list important features or files, this is useful for 
+reviewers to correctly assess how deeply the modifications impact the code base.
+
+For instance:
+
+### New features or files
+- `NewClass` added to `new_file.py`
+- `new_function` added to `existing_file.py`
+
+...
+-->
+
+### New features or files
+
+<!-- List new features or files added. -->
+-
+
+### Modified features or files
+
+<!-- List important modified features or files. -->
+-
+
+### Removed features or files
+
+<!-- List removed features or files. -->
+-
+
+## How has this been tested?
+
+<!-- Describe the tests that you ran to verify your changes. This can be a short 
+description of the tests added to the PR or code snippet to reproduce the change
+in behaviour. -->
+
+
+## Related Issues
+
+<!-- Link to any related issues or discussions. Use keywords like "Fixes", "Resolves",
+or "Closes" to link to issues automatically. -->
+
+- Resolves #
+
+## Breaking changes
+
+<!-- Describe any breaking changes introduced by this PR. -->
+
+## Additional Notes and Examples
+
+<!-- Provide any additional information that will help reviewers understand the
+changes. This can be links to documentations, forum posts, past discussions etc. -->
+
+---
+
+**Please ensure your PR meets the following requirements:**
+
+- [ ] Code builds and passes tests locally, including doctests
+- [ ] New tests have been added (for bug fixes/features)
+- [ ] Pre-commit passes
+- [ ] PR to the documentation exists (for bug fixes / features)

{careamics-0.0.5 → careamics-0.0.7}/.pre-commit-config.yaml

@@ -9,16 +9,16 @@ ci:
 
 repos:
   - repo: https://github.com/abravalheri/validate-pyproject
-    rev: v0.22
+    rev: v0.23
     hooks:
       - id: validate-pyproject
 
   - repo: https://github.com/astral-sh/ruff-pre-commit
-    rev: v0.7.2
+    rev: v0.8.6
     hooks:
       - id: ruff
         exclude: "^src/careamics/lvae_training/.*|^src/careamics/models/lvae/.*|^scripts/.*"
-        args: [--fix, --target-version, py38]
+        args: [--fix, --target-version, py39]
 
   - repo: https://github.com/psf/black
     rev: 24.10.0
@@ -26,14 +26,15 @@ repos:
       - id: black
 
   - repo: https://github.com/pre-commit/mirrors-mypy
-    rev: v1.13.0
+    rev: v1.14.1
     hooks:
       - id: mypy
         files: "^src/"
         exclude: "^src/careamics/lvae_training/.*|^src/careamics/models/lvae/.*|^src/careamics/config/likelihood_model.py|^src/careamics/losses/loss_factory.py|^src/careamics/losses/lvae/losses.py"
         args: ["--config-file", "mypy.ini"]
         additional_dependencies:
-          - numpy
+          - numpy<2.0.0
+          - pydantic
           - types-PyYAML
           - types-setuptools
 
@@ -55,6 +56,6 @@ repos:
 
   # strip out jupyter notebooks
   - repo: https://github.com/kynan/nbstripout
-    rev: 0.8.0
+    rev: 0.8.1
     hooks:
       - id: nbstripout

careamics-0.0.7/CODE_OF_CONDUCT.md

@@ -0,0 +1,128 @@
+# Contributor Covenant Code of Conduct
+
+## Our Pledge
+
+We as members, contributors, and leaders pledge to make participation in our
+community a harassment-free experience for everyone, regardless of age, body
+size, visible or invisible disability, ethnicity, sex characteristics, gender
+identity and expression, level of experience, education, socio-economic status,
+nationality, personal appearance, race, religion, or sexual identity
+and orientation.
+
+We pledge to act and interact in ways that contribute to an open, welcoming,
+diverse, inclusive, and healthy community.
+
+## Our Standards
+
+Examples of behavior that contributes to a positive environment for our
+community include:
+
+* Demonstrating empathy and kindness toward other people
+* Being respectful of differing opinions, viewpoints, and experiences
+* Giving and gracefully accepting constructive feedback
+* Accepting responsibility and apologizing to those affected by our mistakes,
+  and learning from the experience
+* Focusing on what is best not just for us as individuals, but for the
+  overall community
+
+Examples of unacceptable behavior include:
+
+* The use of sexualized language or imagery, and sexual attention or
+  advances of any kind
+* Trolling, insulting or derogatory comments, and personal or political attacks
+* Public or private harassment
+* Publishing others' private information, such as a physical or email
+  address, without their explicit permission
+* Other conduct which could reasonably be considered inappropriate in a
+  professional setting
+
+## Enforcement Responsibilities
+
+Community leaders are responsible for clarifying and enforcing our standards of
+acceptable behavior and will take appropriate and fair corrective action in
+response to any behavior that they deem inappropriate, threatening, offensive,
+or harmful.
+
+Community leaders have the right and responsibility to remove, edit, or reject
+comments, commits, code, wiki edits, issues, and other contributions that are
+not aligned to this Code of Conduct, and will communicate reasons for moderation
+decisions when appropriate.
+
+## Scope
+
+This Code of Conduct applies within all community spaces, and also applies when
+an individual is officially representing the community in public spaces.
+Examples of representing our community include using an official e-mail address,
+posting via an official social media account, or acting as an appointed
+representative at an online or offline event.
+
+## Enforcement
+
+Instances of abusive, harassing, or otherwise unacceptable behavior may be
+reported to the community leaders responsible for enforcement at
+rse@fht.org.
+All complaints will be reviewed and investigated promptly and fairly.
+
+All community leaders are obligated to respect the privacy and security of the
+reporter of any incident.
+
+## Enforcement Guidelines
+
+Community leaders will follow these Community Impact Guidelines in determining
+the consequences for any action they deem in violation of this Code of Conduct:
+
+### 1. Correction
+
+**Community Impact**: Use of inappropriate language or other behavior deemed
+unprofessional or unwelcome in the community.
+
+**Consequence**: A private, written warning from community leaders, providing
+clarity around the nature of the violation and an explanation of why the
+behavior was inappropriate. A public apology may be requested.
+
+### 2. Warning
+
+**Community Impact**: A violation through a single incident or series
+of actions.
+
+**Consequence**: A warning with consequences for continued behavior. No
+interaction with the people involved, including unsolicited interaction with
+those enforcing the Code of Conduct, for a specified period of time. This
+includes avoiding interactions in community spaces as well as external channels
+like social media. Violating these terms may lead to a temporary or
+permanent ban.
+
+### 3. Temporary Ban
+
+**Community Impact**: A serious violation of community standards, including
+sustained inappropriate behavior.
+
+**Consequence**: A temporary ban from any sort of interaction or public
+communication with the community for a specified period of time. No public or
+private interaction with the people involved, including unsolicited interaction
+with those enforcing the Code of Conduct, is allowed during this period.
+Violating these terms may lead to a permanent ban.
+
+### 4. Permanent Ban
+
+**Community Impact**: Demonstrating a pattern of violation of community
+standards, including sustained inappropriate behavior,  harassment of an
+individual, or aggression toward or disparagement of classes of individuals.
+
+**Consequence**: A permanent ban from any sort of public interaction within
+the community.
+
+## Attribution
+
+This Code of Conduct is adapted from the [Contributor Covenant][homepage],
+version 2.0, available at
+https://www.contributor-covenant.org/version/2/0/code_of_conduct.html.
+
+Community Impact Guidelines were inspired by [Mozilla's code of conduct
+enforcement ladder](https://github.com/mozilla/diversity).
+
+[homepage]: https://www.contributor-covenant.org
+
+For answers to common questions about this code of conduct, see the FAQ at
+https://www.contributor-covenant.org/faq. Translations are available at
+https://www.contributor-covenant.org/translations.

{careamics-0.0.5 → careamics-0.0.7}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: careamics
-Version: 0.0.5
+Version: 0.0.7
 Summary: Toolbox for running N2V and friends.
 Project-URL: homepage, https://careamics.github.io/
 Project-URL: repository, https://github.com/CAREamics/careamics
@@ -17,17 +17,19 @@ Classifier: Programming Language :: Python :: 3.12
 Classifier: Typing :: Typed
 Requires-Python: >=3.9
 Requires-Dist: bioimageio-core==0.7
+Requires-Dist: matplotlib<=3.10.0
 Requires-Dist: numpy<2.0.0
-Requires-Dist: pillow<=10.3.0
-Requires-Dist: psutil<=6.1
-Requires-Dist: pydantic<2.9,>=2.5
-Requires-Dist: pytorch-lightning<=2.4,>=2.2
+Requires-Dist: pillow<=11.1.0
+Requires-Dist: psutil<=6.1.1
+Requires-Dist: pydantic<2.11,>=2.5
+Requires-Dist: pytorch-lightning<=2.5.0.post0,>=2.2
 Requires-Dist: pyyaml!=6.0.0,<=6.0.2
-Requires-Dist: scikit-image<=0.23.2
-Requires-Dist: tifffile<=2024.8.30
-Requires-Dist: torch<=2.5,>=2.0
-Requires-Dist: torchvision<=0.20
-Requires-Dist: typer==0.12.3
+Requires-Dist: scikit-image<=0.25.1
+Requires-Dist: tifffile<=2025.1.10
+Requires-Dist: torch<=2.6.0,>=2.0
+Requires-Dist: torchvision<=0.20.1
+Requires-Dist: torchvision<=0.21.0
+Requires-Dist: typer<=0.15.1,>=0.12.3
 Requires-Dist: zarr<3.0.0
 Provides-Extra: dev
 Requires-Dist: onnx; extra == 'dev'
@@ -40,7 +42,7 @@ Requires-Dist: careamics-portfolio; extra == 'examples'
 Requires-Dist: jupyter; extra == 'examples'
 Requires-Dist: matplotlib; extra == 'examples'
 Provides-Extra: tensorboard
-Requires-Dist: protobuf==3.20.3; extra == 'tensorboard'
+Requires-Dist: protobuf==5.29.1; extra == 'tensorboard'
 Requires-Dist: tensorboard; extra == 'tensorboard'
 Provides-Extra: wandb
 Requires-Dist: wandb; extra == 'wandb'

{careamics-0.0.5 → careamics-0.0.7}/mypy.ini

@@ -1,5 +1,6 @@
 [mypy]
 ignore_missing_imports = True
+plugins = pydantic.mypy
 
 [mypy-careamics.lvae_training.*]
 follow_imports = skip

{careamics-0.0.5 → careamics-0.0.7}/pyproject.toml

@@ -43,18 +43,20 @@ classifiers = [
 ]
 dependencies = [
     'numpy<2.0.0',
-    'torch>=2.0,<=2.5',
-    'torchvision<=0.20',
+    'torch>=2.0,<=2.6.0',
+    'torchvision<=0.21.0',
+    'torchvision<=0.20.1',
     'bioimageio.core==0.7',
-    'tifffile<=2024.8.30',
-    'psutil<=6.1',
-    'pydantic>=2.5,<2.9',
-    'pytorch_lightning>=2.2,<=2.4',
+    'tifffile<=2025.1.10',
+    'psutil<=6.1.1',
+    'pydantic>=2.5,<2.11',
+    'pytorch_lightning>=2.2,<=2.5.0.post0',
     'pyyaml<=6.0.2,!=6.0.0',
-    'typer==0.12.3',
-    'scikit-image<=0.23.2',
+    'typer>=0.12.3,<=0.15.1',
+    'scikit-image<=0.25.1',
     'zarr<3.0.0',
-    'pillow<=10.3.0',
+    'pillow<=11.1.0',
+    'matplotlib<=3.10.0'
 ]
 
 [project.optional-dependencies]
@@ -72,7 +74,7 @@ examples = ["jupyter", "careamics-portfolio", "matplotlib"]
 
 # loggers
 wandb = ["wandb"]
-tensorboard = ["tensorboard", "protobuf==3.20.3"]
+tensorboard = ["tensorboard", "protobuf==5.29.1"]
 
 [project.urls]
 homepage = "https://careamics.github.io/"
@@ -190,6 +192,7 @@ checks = [
     "GL02", # Closing quotes should be placed in the line after the last text 
     # in the docstring
     "GL03", # Double line break found
+    "RT04", # Return value description should start with a capital letter
 ]
 exclude = [ # don't report on objects that match any of these regex
     "test_*",

careamics-0.0.7/src/careamics/__init__.py

@@ -0,0 +1,28 @@
+"""Main CAREamics module."""
+
+from importlib.metadata import PackageNotFoundError, version
+
+try:
+    __version__ = version("careamics")
+except PackageNotFoundError:
+    __version__ = "uninstalled"
+
+__all__ = [
+    "CAREamist",
+    "Configuration",
+    "algorithm_factory",
+    "configuration_factory",
+    "data_factory",
+    "load_configuration",
+    "save_configuration",
+]
+
+from .careamist import CAREamist
+from .config import (
+    Configuration,
+    algorithm_factory,
+    configuration_factory,
+    data_factory,
+    load_configuration,
+    save_configuration,
+)

{careamics-0.0.5 → careamics-0.0.7}/src/careamics/careamist.py

@@ -13,7 +13,7 @@ from pytorch_lightning.callbacks import (
 )
 from pytorch_lightning.loggers import CSVLogger, TensorBoardLogger, WandbLogger
 
-from careamics.config import Configuration, FCNAlgorithmConfig, load_configuration
+from careamics.config import Configuration, UNetBasedAlgorithm, load_configuration
 from careamics.config.support import (
     SupportedAlgorithm,
     SupportedArchitecture,
@@ -137,7 +137,7 @@ class CAREamist:
             self.cfg = source
 
             # instantiate model
-            if isinstance(self.cfg.algorithm_config, FCNAlgorithmConfig):
+            if isinstance(self.cfg.algorithm_config, UNetBasedAlgorithm):
                 self.model = FCNModule(
                     algorithm_config=self.cfg.algorithm_config,
                 )
@@ -157,7 +157,8 @@ class CAREamist:
                 self.cfg = load_configuration(source)
 
                 # instantiate model
-                if isinstance(self.cfg.algorithm_config, FCNAlgorithmConfig):
+                # TODO call model factory here
+                if isinstance(self.cfg.algorithm_config, UNetBasedAlgorithm):
                     self.model = FCNModule(
                         algorithm_config=self.cfg.algorithm_config,
                     )  # type: ignore

{careamics-0.0.5 → careamics-0.0.7}/src/careamics/cli/conf.py

@@ -3,12 +3,11 @@
 import sys
 from dataclasses import dataclass
 from pathlib import Path
-from typing import Optional
+from typing import Annotated, Optional
 
 import click
 import typer
 import yaml
-from typing_extensions import Annotated
 
 from ..config import (
     Configuration,

{careamics-0.0.5 → careamics-0.0.7}/src/careamics/cli/main.py

@@ -7,11 +7,10 @@ its implementation is contained in the conf.py file.
 """
 
 from pathlib import Path
-from typing import Optional
+from typing import Annotated, Optional
 
 import click
 import typer
-from typing_extensions import Annotated
 
 from ..careamist import CAREamist
 from . import conf

{careamics-0.0.5 → careamics-0.0.7}/src/careamics/cli/utils.py

@@ -1,11 +1,11 @@
 """Utility functions for the CAREamics CLI."""
 
-from typing import Optional, Tuple
+from typing import Optional
 
 
 def handle_2D_3D_callback(
-    value: Optional[Tuple[int, int, int]]
-) -> Optional[Tuple[int, ...]]:
+    value: Optional[tuple[int, int, int]]
+) -> Optional[tuple[int, ...]]:
     """
     Callback for options that require 2D or 3D inputs.
 

careamics-0.0.7/src/careamics/config/__init__.py

@@ -0,0 +1,63 @@
+"""CAREamics Pydantic configuration models.
+
+To maintain clarity at the module level, we follow the following naming conventions:
+`*_model` is specific for sub-configurations (e.g. architecture, data, algorithm),
+while `*_configuration` is reserved for the main configuration models, including the
+`Configuration` base class and its algorithm-specific child classes.
+"""
+
+__all__ = [
+    "CAREAlgorithm",
+    "CAREConfiguration",
+    "CheckpointModel",
+    "Configuration",
+    "DataConfig",
+    "GaussianMixtureNMConfig",
+    "GeneralDataConfig",
+    "InferenceConfig",
+    "LVAELossConfig",
+    "MultiChannelNMConfig",
+    "N2NAlgorithm",
+    "N2NConfiguration",
+    "N2VAlgorithm",
+    "N2VConfiguration",
+    "N2VDataConfig",
+    "TrainingConfig",
+    "UNetBasedAlgorithm",
+    "VAEBasedAlgorithm",
+    "algorithm_factory",
+    "configuration_factory",
+    "create_care_configuration",
+    "create_n2n_configuration",
+    "create_n2v_configuration",
+    "data_factory",
+    "load_configuration",
+    "save_configuration",
+]
+
+from .algorithms import (
+    CAREAlgorithm,
+    N2NAlgorithm,
+    N2VAlgorithm,
+    UNetBasedAlgorithm,
+    VAEBasedAlgorithm,
+)
+from .callback_model import CheckpointModel
+from .care_configuration import CAREConfiguration
+from .configuration import Configuration
+from .configuration_factories import (
+    algorithm_factory,
+    configuration_factory,
+    create_care_configuration,
+    create_n2n_configuration,
+    create_n2v_configuration,
+    data_factory,
+)
+from .configuration_io import load_configuration, save_configuration
+from .data import DataConfig, GeneralDataConfig, N2VDataConfig
+from .inference_model import InferenceConfig
+from .loss_model import LVAELossConfig
+from .n2n_configuration import N2NConfiguration
+from .n2v_configuration import N2VConfiguration
+from .nm_model import GaussianMixtureNMConfig, MultiChannelNMConfig
+from .training_model import TrainingConfig

careamics-0.0.7/src/careamics/config/algorithms/__init__.py

@@ -0,0 +1,15 @@
+"""Algorithm configurations."""
+
+__all__ = [
+    "CAREAlgorithm",
+    "N2NAlgorithm",
+    "N2VAlgorithm",
+    "UNetBasedAlgorithm",
+    "VAEBasedAlgorithm",
+]
+
+from .care_algorithm_model import CAREAlgorithm
+from .n2n_algorithm_model import N2NAlgorithm
+from .n2v_algorithm_model import N2VAlgorithm
+from .unet_algorithm_model import UNetBasedAlgorithm
+from .vae_algorithm_model import VAEBasedAlgorithm

careamics-0.0.7/src/careamics/config/algorithms/care_algorithm_model.py

@@ -0,0 +1,38 @@
+"""CARE algorithm configuration."""
+
+from typing import Annotated, Literal
+
+from pydantic import AfterValidator
+
+from careamics.config.architectures import UNetModel
+from careamics.config.validators import (
+    model_without_final_activation,
+    model_without_n2v2,
+)
+
+from .unet_algorithm_model import UNetBasedAlgorithm
+
+
+class CAREAlgorithm(UNetBasedAlgorithm):
+    """CARE algorithm configuration.
+
+    Attributes
+    ----------
+    algorithm : "care"
+        CARE Algorithm name.
+    loss : {"mae", "mse"}
+        CARE-compatible loss function.
+    """
+
+    algorithm: Literal["care"] = "care"
+    """CARE Algorithm name."""
+
+    loss: Literal["mae", "mse"] = "mae"
+    """CARE-compatible loss function."""
+
+    model: Annotated[
+        UNetModel,
+        AfterValidator(model_without_n2v2),
+        AfterValidator(model_without_final_activation),
+    ]
+    """UNet without a final activation function and without the `n2v2` modifications."""

careamics-0.0.7/src/careamics/config/algorithms/n2n_algorithm_model.py

@@ -0,0 +1,30 @@
+"""N2N Algorithm configuration."""
+
+from typing import Annotated, Literal
+
+from pydantic import AfterValidator
+
+from careamics.config.architectures import UNetModel
+from careamics.config.validators import (
+    model_without_final_activation,
+    model_without_n2v2,
+)
+
+from .unet_algorithm_model import UNetBasedAlgorithm
+
+
+class N2NAlgorithm(UNetBasedAlgorithm):
+    """Noise2Noise Algorithm configuration."""
+
+    algorithm: Literal["n2n"] = "n2n"
+    """N2N Algorithm name."""
+
+    loss: Literal["mae", "mse"] = "mae"
+    """N2N-compatible loss function."""
+
+    model: Annotated[
+        UNetModel,
+        AfterValidator(model_without_n2v2),
+        AfterValidator(model_without_final_activation),
+    ]
+    """UNet without a final activation function and without the `n2v2` modifications."""

careamics-0.0.7/src/careamics/config/algorithms/n2v_algorithm_model.py

@@ -0,0 +1,29 @@
+""""N2V Algorithm configuration."""
+
+from typing import Annotated, Literal
+
+from pydantic import AfterValidator
+
+from careamics.config.architectures import UNetModel
+from careamics.config.validators import (
+    model_matching_in_out_channels,
+    model_without_final_activation,
+)
+
+from .unet_algorithm_model import UNetBasedAlgorithm
+
+
+class N2VAlgorithm(UNetBasedAlgorithm):
+    """N2V Algorithm configuration."""
+
+    algorithm: Literal["n2v"] = "n2v"
+    """N2V Algorithm name."""
+
+    loss: Literal["n2v"] = "n2v"
+    """N2V loss function."""
+
+    model: Annotated[
+        UNetModel,
+        AfterValidator(model_matching_in_out_channels),
+        AfterValidator(model_without_final_activation),
+    ]