flow-gym-suite 0.1.0__tar.gz

flow_gym_suite-0.1.0/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2025 Antonio Terpin
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

flow_gym_suite-0.1.0/PKG-INFO

@@ -0,0 +1,233 @@
+Metadata-Version: 2.4
+Name: flow-gym-suite
+Version: 0.1.0
+Summary: Flow Gym is a toolkit for research and deployment in flow quantification inspired by OpenAI Gym and Stable-Baselines3.
+Author-email: Antonio Terpin <aterpin@ethz.ch>, Francesco Banelli <fbanelli@ethz.ch>, Alan Bonomi <abonomi@ethz.ch>
+License: MIT License
+        
+        Copyright (c) 2025 Antonio Terpin
+        
+        Permission is hereby granted, free of charge, to any person obtaining a copy
+        of this software and associated documentation files (the "Software"), to deal
+        in the Software without restriction, including without limitation the rights
+        to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+        copies of the Software, and to permit persons to whom the Software is
+        furnished to do so, subject to the following conditions:
+        
+        The above copyright notice and this permission notice shall be included in all
+        copies or substantial portions of the Software.
+        
+        THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+        IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+        FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+        AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+        LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+        OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+        SOFTWARE.
+        
+Project-URL: Homepage, https://github.com/antonioterpin/flowgym
+Project-URL: Repository, https://github.com/antonioterpin/flowgym
+Project-URL: Documentation, https://github.com/antonioterpin/flowgym#readme
+Project-URL: Issues, https://github.com/antonioterpin/flowgym/issues
+Classifier: Development Status :: 4 - Beta
+Classifier: Intended Audience :: Science/Research
+Classifier: Intended Audience :: Developers
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Operating System :: OS Independent
+Classifier: Programming Language :: Python :: 3
+Requires-Python: >=3.10
+Description-Content-Type: text/markdown
+License-File: LICENSE
+Requires-Dist: jax>=0.6.0
+Requires-Dist: flax>=0.10.5
+Requires-Dist: ipywidgets
+Requires-Dist: robo-goggles>=0.1.6
+Requires-Dist: pip>=25.3
+Provides-Extra: other-methods
+Requires-Dist: pyoptflow; extra == "other-methods"
+Requires-Dist: openpiv; extra == "other-methods"
+Requires-Dist: opencv-python-headless; extra == "other-methods"
+Requires-Dist: torch>=1.13.1; extra == "other-methods"
+Requires-Dist: torchvision>=0.14.1; extra == "other-methods"
+Provides-Extra: cuda12
+Requires-Dist: jax[cuda12]>=0.6.0; extra == "cuda12"
+Dynamic: license-file
+
+# 🌊 Flow Gym : A Research Toolkit for Fluid Flow Estimation
+
+[![Python 3.10+](https://img.shields.io/badge/python-3.10%2B-blue.svg)](https://www.python.org/downloads/)
+[![GitHub stars](https://img.shields.io/github/stars/antonioterpin/flowgym?style=social)](https://github.com/antonioterpin/flowgym/stargazers)
+[![PyPI version](https://img.shields.io/pypi/v/flow-gym.svg)](https://pypi.org/project/flow-gym)
+[![License: MIT](https://img.shields.io/badge/License-MIT-yellow.svg)](https://opensource.org/licenses/MIT)
+[![uv](https://img.shields.io/endpoint?url=https://raw.githubusercontent.com/astral-sh/uv/main/assets/badge/v0.json)](https://github.com/astral-sh/uv)
+
+Flow Gym is a **toolkit for research and deployment in flow quantification**, inspired by **OpenAI Gym** and **Stable-Baselines3**.
+It leverages **SynthPix** for synthetic image generation and provides a **unified interface** for testing, training, and deploying learning-based algorithms that estimate fluid flow from sequences of tracer-particle images.
+The framework also includes a growing collection of **integrated algorithms** and **stable JAX re-implementations** of state-of-the-art methods.
+
+![The Flow Gym environment](https://raw.githubusercontent.com/antonioterpin/flowgym/main/docs/flowgym.jpg)
+
+## ✨ Features
+
+- **Stable and reproducible environments**
+  Flow Gym provides **FluidEnv**, a unified environment for both supervised and reinforcement learning. It abstracts away infrastructure details and ensures consistent training and evaluation across experiments.
+
+- **Unified estimator framework**
+  The **Estimator** API makes it easy to integrate, compare, and benchmark different flow estimation algorithms. It also supports extensions beyond velocity estimation — for example, pressure or tracer density — enabling ablation studies and cross-domain applications in fluid dynamics and control.
+
+- **Ready for real-world deployment**
+  Flow Gym includes efficient **JAX implementations** of state-of-the-art flow estimation methods, optimized for real-time performance and suitable for **closed-loop control** or other **fluid-sensing** applications.
+
+## Getting Started 🚀
+
+```bash
+# Basic installation
+uv add flow-gym-suite # or pip install flow-gym-suite
+
+# With CUDA12 support
+uv add "flow-gym-suite[cuda12]"
+```
+
+If you also want to use Flow Gym for training an evaluation, please clone the repository and install the development dependencies as outlined in the [CONTRIBUTING.md](./CONTRIBUTING.md) page.
+There you can also find help if you have any issues installing Flow Gym.
+
+### Basic usage / Deployment  🧭
+
+```python
+from flowgym.make import make_estimator
+# Define the config (or load from YAML)
+model_config = {
+    "estimator": "dis_jax",
+    "estimate_type": "flow",
+    "config": {"jit": True, ..., },
+}
+
+# Create the estimator and associated functions
+trained_state, create_state_fn, compute_estimate_fn, model = make_estimator(
+    model_config=model_config,
+    image_shape=image_shape
+)
+
+# Compute an estimate for an image pair
+est_state = create_state_fn(prev)
+new_est_state, metrics = compute_estimate_fn(curr, est_state)
+```
+
+### Training a model 🧠
+
+```python
+from flowgym.environment.fluid_env import FluidEnv
+
+# Create the environment
+env, env_state = FluidEnv.make(env_config)
+
+for episode in range(num_episodes):
+    obs, env_state, done = env.reset(env_state)
+    est_state = create_state_fn(obs)
+    train_step_fn = model.create_train_step()
+
+    while not done.any():
+        # Estimator forward pass
+        est_state, metrics = compute_estimate_fn(obs, est_state, train_state)
+
+        # Extract the action (the last estimate)
+        action = estimation_state["estimates"][:, -1]
+        # Environment step
+        obs, env_state, reward, done = env.step(env_state, action)
+
+        # Optimization step
+        loss, train_state, _ = train_step_fn(
+            est_state, train_state, reward
+        )
+```
+
+### Implementing a custom estimator 🧩
+
+```python
+from flowgym.base import Estimator
+class MyEstimator(Estimator):
+    def init(self, rng, observation_space, action_space):
+        # Initialize model parameters and state
+        pass
+
+    def _estimate(self, images, state, trainable_state, extras):
+        # Define the forward pass of the model
+        return new_state, extras, metrics
+
+    def create_trainable_state(self, dummy_input, key):
+        # Create the trainable state if needed (e.g., network structure, optimizer, etc.)
+        pass
+
+    def create_train_step(self):
+        def train_step(self, estimation_state, trainable_state, rewards):
+            # Define the training step (loss computation and optimization)
+            return loss, new_trainable_state, metrics
+```
+
+## Custom Preprocessing  🧼
+
+```python
+# flowgym/common/preprocess.py
+def custom_preprocess_validate_params(params: Dict[str, Any]) -> None:
+    # Validate custom pre-processing parameters
+    pass
+
+def custom_preprocess(
+    images: jnp.ndarray,
+    params: Dict[str, Any]
+) -> jnp.ndarray:
+    # Implement custom pre-processing logic
+    return processed_images
+
+__all__ = [
+    "custom_preprocess_validate_params",
+    "custom_preprocess",
+]
+```
+
+## Custom Postprocessing  🎨
+
+```python
+# flowgym/flow/postprocess/__init__.py
+def custom_flow_postprocess_validate_params(params: Dict[str, Any]) -> None:
+    # Validate custom post-processing parameters
+    pass
+
+def custom_flow_postprocess(
+    estimates: jnp.ndarray,
+    params: Dict[str, Any]
+) -> jnp.ndarray:
+    # Implement custom flow post-processing logic
+    return processed_estimates
+
+__all__ = [
+    "custom_flow_postprocess_validate_params",
+    "custom_flow_postprocess",
+]
+```
+
+## Examples 📚
+
+For more examples, please check our [training](https://raw.githubusercontent.com/antonioterpin/flowgym/main/src/flowgym/train.py) and [evaluation](https://raw.githubusercontent.com/antonioterpin/flowgym/main/src/flowgym/eval.py) scripts.
+
+## Contributing 🤗
+
+Contributions are more than welcome! 🙏 Please check out our [how to contribute page](https://raw.githubusercontent.com/antonioterpin/flowgym/main/CONTRIBUTING.md), and feel free to open an issue for problems and feature requests⚠️.
+
+If you have developed a new method integrating with Flow Gym, please do open a PR! New methods should come with reproducible experiments showcasing the features. We collect these experiments in the folder `experiments/your-method`.
+
+## Citation 📈
+
+If you use this code in your research, please cite our paper:
+
+```bibtex
+   @article{banelli2025flowgym,
+      title={Flow Gym},
+      author={Banelli, Francesco and Terpin, Antonio and Bonomi, Alan and D'Andrea, Raffaello},
+      year={2025}
+   }
+```
+
+## License 📝
+
+This project is licensed under the MIT License - see the [LICENSE](https://raw.githubusercontent.com/antonioterpin/flowgym/main/LICENSE) file for details.

flow_gym_suite-0.1.0/README.md

@@ -0,0 +1,178 @@
+# 🌊 Flow Gym : A Research Toolkit for Fluid Flow Estimation
+
+[![Python 3.10+](https://img.shields.io/badge/python-3.10%2B-blue.svg)](https://www.python.org/downloads/)
+[![GitHub stars](https://img.shields.io/github/stars/antonioterpin/flowgym?style=social)](https://github.com/antonioterpin/flowgym/stargazers)
+[![PyPI version](https://img.shields.io/pypi/v/flow-gym.svg)](https://pypi.org/project/flow-gym)
+[![License: MIT](https://img.shields.io/badge/License-MIT-yellow.svg)](https://opensource.org/licenses/MIT)
+[![uv](https://img.shields.io/endpoint?url=https://raw.githubusercontent.com/astral-sh/uv/main/assets/badge/v0.json)](https://github.com/astral-sh/uv)
+
+Flow Gym is a **toolkit for research and deployment in flow quantification**, inspired by **OpenAI Gym** and **Stable-Baselines3**.
+It leverages **SynthPix** for synthetic image generation and provides a **unified interface** for testing, training, and deploying learning-based algorithms that estimate fluid flow from sequences of tracer-particle images.
+The framework also includes a growing collection of **integrated algorithms** and **stable JAX re-implementations** of state-of-the-art methods.
+
+![The Flow Gym environment](https://raw.githubusercontent.com/antonioterpin/flowgym/main/docs/flowgym.jpg)
+
+## ✨ Features
+
+- **Stable and reproducible environments**
+  Flow Gym provides **FluidEnv**, a unified environment for both supervised and reinforcement learning. It abstracts away infrastructure details and ensures consistent training and evaluation across experiments.
+
+- **Unified estimator framework**
+  The **Estimator** API makes it easy to integrate, compare, and benchmark different flow estimation algorithms. It also supports extensions beyond velocity estimation — for example, pressure or tracer density — enabling ablation studies and cross-domain applications in fluid dynamics and control.
+
+- **Ready for real-world deployment**
+  Flow Gym includes efficient **JAX implementations** of state-of-the-art flow estimation methods, optimized for real-time performance and suitable for **closed-loop control** or other **fluid-sensing** applications.
+
+## Getting Started 🚀
+
+```bash
+# Basic installation
+uv add flow-gym-suite # or pip install flow-gym-suite
+
+# With CUDA12 support
+uv add "flow-gym-suite[cuda12]"
+```
+
+If you also want to use Flow Gym for training an evaluation, please clone the repository and install the development dependencies as outlined in the [CONTRIBUTING.md](./CONTRIBUTING.md) page.
+There you can also find help if you have any issues installing Flow Gym.
+
+### Basic usage / Deployment  🧭
+
+```python
+from flowgym.make import make_estimator
+# Define the config (or load from YAML)
+model_config = {
+    "estimator": "dis_jax",
+    "estimate_type": "flow",
+    "config": {"jit": True, ..., },
+}
+
+# Create the estimator and associated functions
+trained_state, create_state_fn, compute_estimate_fn, model = make_estimator(
+    model_config=model_config,
+    image_shape=image_shape
+)
+
+# Compute an estimate for an image pair
+est_state = create_state_fn(prev)
+new_est_state, metrics = compute_estimate_fn(curr, est_state)
+```
+
+### Training a model 🧠
+
+```python
+from flowgym.environment.fluid_env import FluidEnv
+
+# Create the environment
+env, env_state = FluidEnv.make(env_config)
+
+for episode in range(num_episodes):
+    obs, env_state, done = env.reset(env_state)
+    est_state = create_state_fn(obs)
+    train_step_fn = model.create_train_step()
+
+    while not done.any():
+        # Estimator forward pass
+        est_state, metrics = compute_estimate_fn(obs, est_state, train_state)
+
+        # Extract the action (the last estimate)
+        action = estimation_state["estimates"][:, -1]
+        # Environment step
+        obs, env_state, reward, done = env.step(env_state, action)
+
+        # Optimization step
+        loss, train_state, _ = train_step_fn(
+            est_state, train_state, reward
+        )
+```
+
+### Implementing a custom estimator 🧩
+
+```python
+from flowgym.base import Estimator
+class MyEstimator(Estimator):
+    def init(self, rng, observation_space, action_space):
+        # Initialize model parameters and state
+        pass
+
+    def _estimate(self, images, state, trainable_state, extras):
+        # Define the forward pass of the model
+        return new_state, extras, metrics
+
+    def create_trainable_state(self, dummy_input, key):
+        # Create the trainable state if needed (e.g., network structure, optimizer, etc.)
+        pass
+
+    def create_train_step(self):
+        def train_step(self, estimation_state, trainable_state, rewards):
+            # Define the training step (loss computation and optimization)
+            return loss, new_trainable_state, metrics
+```
+
+## Custom Preprocessing  🧼
+
+```python
+# flowgym/common/preprocess.py
+def custom_preprocess_validate_params(params: Dict[str, Any]) -> None:
+    # Validate custom pre-processing parameters
+    pass
+
+def custom_preprocess(
+    images: jnp.ndarray,
+    params: Dict[str, Any]
+) -> jnp.ndarray:
+    # Implement custom pre-processing logic
+    return processed_images
+
+__all__ = [
+    "custom_preprocess_validate_params",
+    "custom_preprocess",
+]
+```
+
+## Custom Postprocessing  🎨
+
+```python
+# flowgym/flow/postprocess/__init__.py
+def custom_flow_postprocess_validate_params(params: Dict[str, Any]) -> None:
+    # Validate custom post-processing parameters
+    pass
+
+def custom_flow_postprocess(
+    estimates: jnp.ndarray,
+    params: Dict[str, Any]
+) -> jnp.ndarray:
+    # Implement custom flow post-processing logic
+    return processed_estimates
+
+__all__ = [
+    "custom_flow_postprocess_validate_params",
+    "custom_flow_postprocess",
+]
+```
+
+## Examples 📚
+
+For more examples, please check our [training](https://raw.githubusercontent.com/antonioterpin/flowgym/main/src/flowgym/train.py) and [evaluation](https://raw.githubusercontent.com/antonioterpin/flowgym/main/src/flowgym/eval.py) scripts.
+
+## Contributing 🤗
+
+Contributions are more than welcome! 🙏 Please check out our [how to contribute page](https://raw.githubusercontent.com/antonioterpin/flowgym/main/CONTRIBUTING.md), and feel free to open an issue for problems and feature requests⚠️.
+
+If you have developed a new method integrating with Flow Gym, please do open a PR! New methods should come with reproducible experiments showcasing the features. We collect these experiments in the folder `experiments/your-method`.
+
+## Citation 📈
+
+If you use this code in your research, please cite our paper:
+
+```bibtex
+   @article{banelli2025flowgym,
+      title={Flow Gym},
+      author={Banelli, Francesco and Terpin, Antonio and Bonomi, Alan and D'Andrea, Raffaello},
+      year={2025}
+   }
+```
+
+## License 📝
+
+This project is licensed under the MIT License - see the [LICENSE](https://raw.githubusercontent.com/antonioterpin/flowgym/main/LICENSE) file for details.

flow_gym_suite-0.1.0/pyproject.toml

@@ -0,0 +1,210 @@
+[build-system]
+requires = ["setuptools>=61.0", "wheel"]
+build-backend = "setuptools.build_meta"
+
+[project]
+name = "flow-gym-suite"
+version = "0.1.0"
+description = "Flow Gym is a toolkit for research and deployment in flow quantification inspired by OpenAI Gym and Stable-Baselines3."
+readme = "README.md"
+license = { file = "LICENSE" }
+authors = [
+    { name = "Antonio Terpin", email = "aterpin@ethz.ch" },
+    { name = "Francesco Banelli", email = "fbanelli@ethz.ch" },
+    { name = "Alan Bonomi", email = "abonomi@ethz.ch" }
+]
+classifiers = [
+    "Development Status :: 4 - Beta",
+    "Intended Audience :: Science/Research",
+    "Intended Audience :: Developers",
+    "License :: OSI Approved :: MIT License",
+    "Operating System :: OS Independent",
+    "Programming Language :: Python :: 3",
+]
+requires-python = ">=3.10"
+
+# Core dependencies
+dependencies = [
+    "jax>=0.6.0",
+    "flax>=0.10.5",
+    "ipywidgets",
+    "robo-goggles>=0.1.6",
+    "pip>=25.3",
+]
+
+[project.urls]
+Homepage = "https://github.com/antonioterpin/flowgym"
+Repository = "https://github.com/antonioterpin/flowgym"
+Documentation = "https://github.com/antonioterpin/flowgym#readme"
+Issues = "https://github.com/antonioterpin/flowgym/issues"
+
+[tool.setuptools.packages.find]
+where = ["src"]
+include = ["flowgym*"]
+exclude = ["tests*", "docs*"]
+
+# Dependency groups
+[dependency-groups]
+dev = [
+    "synthpix",
+    "pre_commit==4.0.1",
+    "snowballstemmer",
+    "pytest==7.4.4",
+    "wandb",
+    "scikit-image>=0.20.0",
+]
+
+synthpix-cuda12 = [
+    "synthpix[cuda12]",
+]
+
+dev-cuda12 = [
+    {include-group = "dev"},
+    {include-group = "synthpix-cuda12"},
+]
+
+docs = [
+    "Sphinx==7.4.7",
+    "sphinx-copybutton==0.5.2",
+    "sphinx-rtd-theme==2.0.0",
+    "sphinx-tabs==3.4.7",
+    "sphinx-togglebutton==0.3.2",
+    "sphinxcontrib-applehelp==2.0.0",
+    "sphinxcontrib-bibtex==2.6.2",
+    "sphinxcontrib-devhelp==2.0.0",
+    "sphinxcontrib-htmlhelp==2.1.0",
+    "sphinxcontrib-jquery==4.1",
+    "sphinxcontrib-jsmath==1.0.1",
+    "sphinxcontrib-qthelp==2.0.0",
+    "sphinxcontrib-serializinghtml==2.0.0",
+]
+
+# Optional dependencies
+[project.optional-dependencies]
+other_methods = [
+    "pyoptflow",
+    "openpiv",
+    "opencv-python-headless",
+    "torch>=1.13.1",
+    "torchvision>=0.14.1",
+]
+
+cuda12 = [
+    "jax[cuda12]>=0.6.0",
+]
+
+# autopep8 configuration for 4-space indentation
+[tool.autopep8]
+max_line_length = 80
+indent_size = 4
+aggressive = 2
+in_place = true
+exclude = ".eggs,.git,.hg,.mypy_cache,.tox,.venv,_build,buck-out,build,dist"
+
+# isort configuration (updated for autopep8 compatibility)
+[tool.isort]
+line_length = 80
+indent = 4
+multi_line_output = 0
+force_single_line = false
+include_trailing_comma = false
+force_grid_wrap = 0
+use_parentheses = true
+ensure_newline_before_comments = true
+
+# MyPy configuration
+[tool.mypy]
+python_version = "3.10"
+warn_return_any = true
+warn_unused_configs = true
+disallow_untyped_defs = true
+disallow_incomplete_defs = true
+check_untyped_defs = true
+disallow_untyped_decorators = true
+no_implicit_optional = true
+warn_redundant_casts = true
+warn_unused_ignores = true
+warn_no_return = true
+warn_unreachable = true
+strict_equality = true
+show_error_codes = true
+explicit_package_bases = true
+
+# Allow untyped decorators in test files for Hypothesis
+[[tool.mypy.overrides]]
+module = "tests.*"
+disallow_untyped_decorators = false
+
+# Pydocstyle configuration
+[tool.pydocstyle]
+convention = "google"
+match-dir = "^(?!tests).*"
+
+# Bandit security configuration
+[tool.bandit]
+exclude_dirs = ["tests", ".venv", "build", "dist"]
+
+# Pytest configuration
+[tool.pytest.ini_options]
+minversion = "7.0"
+addopts = [
+        "-ra",
+        "--strict-markers",
+        "--strict-config",
+        "--cov=goggles",
+        "--cov-branch",
+        "--cov-report=term-missing",
+        "--cov-report=html",
+        "--cov-report=xml",
+]
+testpaths = ["tests"]
+filterwarnings = [
+        "error",
+        "ignore::UserWarning",
+        "ignore::DeprecationWarning",
+]
+
+# Ruff configuration for linting and modern typing enforcement
+[tool.ruff]
+line-length = 80
+target-version = "py310"
+
+[tool.ruff.lint]
+# Enable comprehensive rule sets
+select = [
+    "E",      # pycodestyle errors
+    "W",      # pycodestyle warnings
+    "F",      # pyflakes (including unused imports F401)
+    "UP",     # pyupgrade (modern typing patterns)
+    "I",      # isort
+    "N",      # pep8-naming
+    "D",      # pydocstyle
+    "PL",     # pylint
+    "RUF",    # ruff-specific rules
+]
+
+ignore = [
+    "D100",   # Missing docstring in public module
+    "D104",   # Missing docstring in public package
+    "D105",   # Missing docstring in magic method
+    "PLR0913", # Too many arguments to function call
+    "PLR2004", # Magic value used in comparison
+]
+
+# Specific rules for modern typing enforcement
+[tool.ruff.lint.pyupgrade]
+# Keep this empty to use all pyupgrade rules
+
+[tool.ruff.lint.per-file-ignores]
+"tests/*" = ["D", "PLR2004"]  # Ignore docstring rules in tests
+"examples/*" = ["D"]          # Ignore docstring rules in examples
+
+[tool.ruff.lint.pydocstyle]
+convention = "google"
+
+[tool.setuptools.dynamic]
+version = {attr = "flowgym.__version__"}
+
+[tool.uv]
+default-groups = []
+ 

flow_gym_suite-0.1.0/setup.cfg

@@ -0,0 +1,4 @@
+[egg_info]
+tag_build = 
+tag_date = 0
+