moospread 0.1.1__tar.gz → 0.1.3__tar.gz

{moospread-0.1.1/moospread.egg-info → moospread-0.1.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: moospread
-Version: 0.1.1
+Version: 0.1.3
 Summary: Sampling-based Pareto front Refinement via Efficient Adaptive Diffusion
 Author-email: Sedjro Salomon Hotegni <salomon.hotegni@aims.ac.rw>
 Maintainer-email: Sedjro Salomon Hotegni <salomon.hotegni@tu-dortmund.de>
@@ -21,13 +21,36 @@ Project-URL: Homepage, https://github.com/safe-autonomous-systems/moo-spread
 Requires-Python: >=3.8
 Description-Content-Type: text/markdown
 License-File: LICENSE
-Requires-Dist: numpy
-Requires-Dist: torch
+Requires-Dist: numpy>=1.24
+Requires-Dist: scipy>=1.10
+Requires-Dist: scikit-learn>=1.3
+Requires-Dist: matplotlib>=3.7
+Requires-Dist: pandas>=2.0
+Requires-Dist: importlib
+Requires-Dist: opencv-python-headless
+Requires-Dist: pytz
+Requires-Dist: PyYAML>=6.0
+Requires-Dist: tqdm>=4.66
+Requires-Dist: cvxopt>=1.3.2
+Requires-Dist: pymoo>=0.5
+Requires-Dist: torch>=2.2
+Requires-Dist: torchvision
+Requires-Dist: torchaudio
+Requires-Dist: gpytorch>=1.10
+Requires-Dist: botorch>=0.8.5
+Requires-Dist: pyro-ppl>=1.9.1
 Provides-Extra: dev
-Requires-Dist: coverage; extra == "dev"
-Requires-Dist: mypy; extra == "dev"
 Requires-Dist: pytest; extra == "dev"
 Requires-Dist: ruff; extra == "dev"
+Requires-Dist: build; extra == "dev"
+Requires-Dist: twine; extra == "dev"
+Provides-Extra: plot
+Requires-Dist: seaborn>=0.13; extra == "plot"
+Provides-Extra: config
+Requires-Dist: hydra-core>=1.3.2; extra == "config"
+Requires-Dist: omegaconf>=2.3.0; extra == "config"
+Provides-Extra: tracking
+Requires-Dist: wandb>=0.20; extra == "tracking"
 Dynamic: license-file
 
 <p align="center">
@@ -52,8 +75,6 @@ Dynamic: license-file
 conda create -n moospread python=3.11
 conda activate moospread
 pip install moospread
-# To install CUDA‐enabled PyTorch, run (or visit: https://pytorch.org/get-started/locally/):
-pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118
 ```
 Or, to install the latest code from GitHub:
 ```python
@@ -62,8 +83,6 @@ conda activate moospread
 git clone https://github.com/safe-autonomous-systems/moo-spread.git
 cd moo-spread
 pip install -e .
-# To install CUDA‐enabled PyTorch, run (or visit: https://pytorch.org/get-started/locally/):
-pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118
 ```
 ### Basic usage
 This example shows how to solve a standard multi-objective optimization benchmark (ZDT2) using the **SPREAD** solver.
@@ -79,35 +98,23 @@ from moospread import SPREAD
 from moospread.tasks import ZDT2
 
 # Define the problem
-n_var = 30
-problem = ZDT2(n_var=n_var)
+problem = ZDT2(n_var=30)
 
 # Initialize the SPREAD solver
 solver = SPREAD(
     problem,
     data_size=10000,
-    timesteps=5000,
+    timesteps=1000,
     num_epochs=1000,
     train_tol=100,
-    num_blocks=3,
-    validation_split=0.1,
     mode="online",
     seed=2026,
     verbose=True
 )
 
 # Solve the problem
-results = solver.solve(
+res_x, res_y = solver.solve(
     num_points_sample=200,
-    strict_guidance=False,
-    rho_scale_gamma=0.9,
-    nu_t=10.0,
-    eta_init=0.9,
-    num_inner_steps=10,
-    lr_inner=0.9,
-    free_initial_h=True,
-    use_sigma_rep=False,
-    kernel_sigma_rep=0.01,
     iterative_plot=True,
     plot_period=10,
     max_backtracks=25,

{moospread-0.1.1 → moospread-0.1.3}/README.md

@@ -1,16 +1,28 @@
 <p align="center">
   <img src="/images/logo_well_spread.png" style="width: 30%; height: auto;">
 </p>
-<a href="https://pypi.org/project/moospread/"><img src="https://img.shields.io/pypi/v/advermorel.svg" alt="PyPI version"></a>
-
-# [ICLR 2026] SPREAD: Sampling-based Pareto front Refinement via Efficient Adaptive Diffusion
-
 <!--
+[![PyPI version](https://badge.fury.io/py/moospread.svg)](https://badge.fury.io/py/moospread)
+-->
 <p align="center">
-<a href="https://pypi.org/project/moospread/"><img src="https://img.shields.io/pypi/v/advermorel.svg" alt="PyPI version"></a>
+  <a href="https://pypi.org/project/moospread/"><img src="https://img.shields.io/pypi/v/moospread.svg" alt="PyPI version"></a>
+  <a href="https://moospread.readthedocs.io">
+  <img src="https://img.shields.io/badge/docs-online-brightgreen.svg" alt="Documentation">
+</a>
 </p>
+<div align="center">
+    <h3>
+      <a href="https://pypi.org/project/moospread/">Installation</a> |
+      <a href="https://moospread.readthedocs.io/en/latest/">Documentation</a> | 
+      <a href="https://arxiv.org/pdf/2509.21058">Paper</a>
+    </h3>
+</div>
+<!--
+<a href="https://pypi.org/project/moospread/"><img src="https://img.shields.io/pypi/v/advermorel.svg" alt="PyPI version"></a>
 -->
 
+# [ICLR 2026] SPREAD: Sampling-based Pareto front Refinement via Efficient Adaptive Diffusion
+
 > SPREAD is a novel sampling-based approach for multi-objective optimization that leverages diffusion models to efficiently refine and generate well-spread Pareto front approximations. It combines the expressiveness of diffusion models with multi-objective optimization principles to achieve both high convergence to the Pareto front and excellent diversity across the objective space. SPREAD demonstrates competitive performance against state-of-the-art methods while providing a flexible framework for different optimization contexts.
 
 ## 🚀 Getting Started
@@ -21,8 +33,6 @@
 conda create -n moospread python=3.11
 conda activate moospread
 pip install moospread
-# To install CUDA‐enabled PyTorch, run (or visit: https://pytorch.org/get-started/locally/):
-pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118
 ```
 Or, to install the latest code from GitHub:
 ```python
@@ -31,8 +41,6 @@ conda activate moospread
 git clone https://github.com/safe-autonomous-systems/moo-spread.git
 cd moo-spread
 pip install -e .
-# To install CUDA‐enabled PyTorch, run (or visit: https://pytorch.org/get-started/locally/):
-pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118
 ```
 ### Basic usage
 This example shows how to solve a standard multi-objective optimization benchmark (ZDT2) using the **SPREAD** solver.
@@ -48,35 +56,23 @@ from moospread import SPREAD
 from moospread.tasks import ZDT2
 
 # Define the problem
-n_var = 30
-problem = ZDT2(n_var=n_var)
+problem = ZDT2(n_var=30)
 
 # Initialize the SPREAD solver
 solver = SPREAD(
     problem,
     data_size=10000,
-    timesteps=5000,
+    timesteps=1000,
     num_epochs=1000,
     train_tol=100,
-    num_blocks=3,
-    validation_split=0.1,
     mode="online",
     seed=2026,
     verbose=True
 )
 
 # Solve the problem
-results = solver.solve(
+res_x, res_y = solver.solve(
     num_points_sample=200,
-    strict_guidance=False,
-    rho_scale_gamma=0.9,
-    nu_t=10.0,
-    eta_init=0.9,
-    num_inner_steps=10,
-    lr_inner=0.9,
-    free_initial_h=True,
-    use_sigma_rep=False,
-    kernel_sigma_rep=0.01,
     iterative_plot=True,
     plot_period=10,
     max_backtracks=25,
@@ -90,11 +86,10 @@ This will train a diffusion-based multi-objective solver, approximate the Pareto
 
 ---
 
-<!--
+
 ### 📚 Next steps
 
-For more advanced examples (offline mode, Bayesian mode, custom problems), see the full [documentation](https://moospread.readthedocs.io/en/latest/).
--->
+For more advanced examples (offline mode, mobo mode, tutorials), see the full [documentation](https://moospread.readthedocs.io/en/latest/).
 
 ## 🔬 Experiments
 

{moospread-0.1.1 → moospread-0.1.3}/README_PYPI.md

@@ -20,8 +20,6 @@
 conda create -n moospread python=3.11
 conda activate moospread
 pip install moospread
-# To install CUDA‐enabled PyTorch, run (or visit: https://pytorch.org/get-started/locally/):
-pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118
 ```
 Or, to install the latest code from GitHub:
 ```python
@@ -30,8 +28,6 @@ conda activate moospread
 git clone https://github.com/safe-autonomous-systems/moo-spread.git
 cd moo-spread
 pip install -e .
-# To install CUDA‐enabled PyTorch, run (or visit: https://pytorch.org/get-started/locally/):
-pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118
 ```
 ### Basic usage
 This example shows how to solve a standard multi-objective optimization benchmark (ZDT2) using the **SPREAD** solver.
@@ -47,35 +43,23 @@ from moospread import SPREAD
 from moospread.tasks import ZDT2
 
 # Define the problem
-n_var = 30
-problem = ZDT2(n_var=n_var)
+problem = ZDT2(n_var=30)
 
 # Initialize the SPREAD solver
 solver = SPREAD(
     problem,
     data_size=10000,
-    timesteps=5000,
+    timesteps=1000,
     num_epochs=1000,
     train_tol=100,
-    num_blocks=3,
-    validation_split=0.1,
     mode="online",
     seed=2026,
     verbose=True
 )
 
 # Solve the problem
-results = solver.solve(
+res_x, res_y = solver.solve(
     num_points_sample=200,
-    strict_guidance=False,
-    rho_scale_gamma=0.9,
-    nu_t=10.0,
-    eta_init=0.9,
-    num_inner_steps=10,
-    lr_inner=0.9,
-    free_initial_h=True,
-    use_sigma_rep=False,
-    kernel_sigma_rep=0.01,
     iterative_plot=True,
     plot_period=10,
     max_backtracks=25,