paddle 1.1.1__tar.gz → 1.1.3__tar.gz

paddle-1.1.3/.bumpversion.cfg

@@ -0,0 +1,12 @@
+[bumpversion]
+current_version = 1.1.3
+commit = True
+tag = True
+
+[bumpversion:file:src/paddle/__init__.py]
+search = __version__ = "{current_version}"
+replace = __version__ = "{new_version}"
+
+[bumpversion:file:pyproject.toml]
+search = version = "{current_version}"
+replace = version = "{new_version}"

paddle-1.1.3/.github/workflows/bump-and-tag.yaml

@@ -0,0 +1,78 @@
+name: Bump Version and Tag
+
+on:
+  pull_request:
+    types: [closed]
+    branches: [main]
+
+env:
+  PYTHON_VERSION: "3.11"
+
+jobs:
+  bump-version:
+    runs-on: ubuntu-latest
+    permissions:
+      contents: write
+      pull-requests: write
+
+    steps:
+      - name: Checkout repository
+        uses: actions/checkout@v4
+        with:
+          fetch-depth: 0
+
+      - name: Set up Python
+        uses: actions/setup-python@v5
+        with:
+          python-version: ${{ env.PYTHON_VERSION }}
+
+      - name: Configure Git
+        run: |
+          git config user.name "github-actions[bot]"
+          git config user.email "github-actions[bot]@users.noreply.github.com"
+
+      - name: Bump version and push tag
+        id: bump_version
+        uses: jasonamyers/github-bumpversion-action@v1.0.5
+        env:
+          GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }}
+
+      - name: Push changes
+        uses: ad-m/github-push-action@master
+        with:
+          github_token: ${{ secrets.GITHUB_TOKEN }}
+          branch: ${{ github.ref }}
+          tags: true
+
+      - name: Generate Release Notes
+        uses: octokit/request-action@v2.x
+        id: get_release_notes
+        with:
+          route: POST /repos/${{ github.repository }}/releases/generate-notes
+          tag_name: v${{ steps.bump_version.outputs.new_ver }}
+        env:
+          GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }}
+
+      - name: Create GitHub release
+        id: create_release
+        uses: actions/create-release@v1.1.4
+        env:
+          GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }}
+        with:
+          tag_name: v${{ steps.bump_version.outputs.new_ver }}
+          release_name: Release v${{ steps.bump_version.outputs.new_ver }}
+          draft: false
+          prerelease: false
+          body: ${{ fromJson(steps.get_release_notes.outputs.data).body }}
+
+      - name: Post comment on PR with release notes
+        uses: actions/github-script@v7
+        with:
+          script: |
+            const notes = `${{ fromJson(steps.get_release_notes.outputs.data).body }}`;
+            github.rest.issues.createComment({
+              issue_number: context.payload.pull_request.number,
+              owner: context.repo.owner,
+              repo: context.repo.repo,
+              body: `🎉 Released v${{ steps.bump_version.outputs.new_ver }}!\n\n${notes}`
+            })

paddle-1.1.3/.github/workflows/cd.yml

@@ -0,0 +1,55 @@
+name: Deploy to PyPI
+
+on:
+  push:
+    branches: [main]
+    tags: [v*]
+
+jobs:
+  build:
+    name: Build distribution artifacts
+    runs-on: ubuntu-latest
+    steps:
+      - name: Checkout code
+        uses: actions/checkout@v4
+
+      - name: Setup Host Python
+        uses: actions/setup-python@v5
+        with:
+          python-version: "3.11"
+          cache: pip
+
+      - name: Install build backend
+        run: |
+          python -m pip install --upgrade pip
+          pip install build
+
+      - name: Build sdist and wheel
+        run: python -m build
+
+      - name: Upload artifacts
+        uses: actions/upload-artifact@v4
+        with:
+          name: dist
+          path: dist/*
+
+  publish-pypi:
+    name: Publish to PyPI
+    needs: build
+    runs-on: ubuntu-latest
+    permissions:
+      id-token: write
+      contents: read
+    steps:
+      - name: Download wheels
+        uses: actions/download-artifact@v4
+        with:
+          name: dist
+          path: dist
+
+      - name: Publish to PyPI
+        uses: pypa/gh-action-pypi-publish@release/v1
+        with:
+          user: __token__
+          password: ${{ secrets.PYPI_API_TOKEN }}
+          verbose: true

paddle-1.1.3/.github/workflows/ci.yml

@@ -0,0 +1,60 @@
+name: Continuous Integration
+
+on:
+  pull_request:
+    branches: [main]
+
+  push:
+    branches: [main]
+    tags: [v*]
+
+env:
+  PYTHON_VERSION: "3.11"
+  BUILD_TYPE: Release
+
+jobs:
+  pre-commit:
+    runs-on: ubuntu-latest
+    steps:
+      - name: Check out code
+        uses: actions/checkout@v4
+
+      - name: Set up Python
+        uses: actions/setup-python@v5
+        with:
+          python-version: ${{ env.PYTHON_VERSION }}
+
+      - name: Install style checkers
+        run: pip install --user cpplint cppcheck clang-format==20.1.4
+
+      - name: Cache pre-commit
+        uses: actions/cache@v3
+        with:
+          path: ~/.cache/pre-commit
+          key: ${{ runner.os }}-pre-commit-${{ hashFiles('.pre-commit-config.yaml') }}
+          restore-keys: |
+            ${{ runner.os }}-pre-commit-
+
+      - name: Run pre-commit
+        uses: pre-commit/action@v3.0.1
+
+  build-and-test:
+    needs: pre-commit
+    runs-on: ${{ matrix.os }}
+    strategy:
+      fail-fast: true
+      matrix:
+        os: [ubuntu-latest, macOS-latest]
+    steps:
+      - name: Check out code
+        uses: actions/checkout@v4
+
+      - name: Set up Python
+        uses: actions/setup-python@v5
+        with:
+          python-version: ${{ env.PYTHON_VERSION }}
+          cache: "pip"
+
+      - name: Install Python dependencies
+        run: |
+          pip install numpy pytest 'pyharp>=1.7.1' 'torch==2.7.1'

{paddle-1.1.1 → paddle-1.1.3}/.gitignore

@@ -11,4 +11,3 @@ htmlcov/
 .venv/
 .env
 *.log
-

paddle-1.1.3/.pre-commit-config.yaml

@@ -0,0 +1,17 @@
+repos:
+  - repo: https://github.com/pre-commit/pre-commit-hooks
+    rev: v4.0.1
+    hooks:
+      - id: requirements-txt-fixer
+      - id: trailing-whitespace
+        exclude: ^(data|patches)/.*$
+      - id: end-of-file-fixer
+        exclude: ^(data|patches)/.*$
+      - id: check-yaml
+        exclude: ^data/
+
+  - repo: https://github.com/psf/black
+    rev: 24.10.0  # Use the latest stable version
+    hooks:
+      - id: black
+        args: [--line-length=88]

{paddle-1.1.1 → paddle-1.1.3}/PKG-INFO

@@ -1,11 +1,11 @@
 Metadata-Version: 2.4
 Name: paddle
-Version: 1.1.1
+Version: 1.1.3
 Summary: Canoe's utility subroutines
-Project-URL: Homepage, https://github.com/chengcli/paddle
-Project-URL: Repository, https://github.com/chengcli/paddle
-Project-URL: Issues, https://github.com/chengcli/paddle/issues
-Author-email: Cheng Li <chengcli@umich.edu>
+Project-URL: Homepage, https://github.com/elijah-mullens/paddle
+Project-URL: Repository, https://github.com/elijah-mullens/paddle
+Project-URL: Issues, https://github.com/elijah-mullens/paddle/issues
+Author-email: Cheng Li <chengcli@umich.edu>, Elijah Mullens <eem85@cornell.edu>
 Classifier: Development Status :: 3 - Alpha
 Classifier: Intended Audience :: Developers
 Classifier: Intended Audience :: Science/Research

{paddle-1.1.1 → paddle-1.1.3}/pyproject.toml

@@ -4,11 +4,14 @@ build-backend = "hatchling.build"
 
 [project]
 name = "paddle"
-version = "1.1.1"
+version = "1.1.3"
 description = "Canoe's utility subroutines"
 readme = "README.md"
 requires-python = ">=3.9"
-authors = [{ name = "Cheng Li", email = "chengcli@umich.edu" }]
+authors = [
+  { name = "Cheng Li", email = "chengcli@umich.edu" },
+  { name = "Elijah Mullens", email = "eem85@cornell.edu" },
+]
 keywords = []
 classifiers = [
   "Development Status :: 3 - Alpha",
@@ -39,9 +42,9 @@ dev = [
 ]
 
 [project.urls]
-Homepage = "https://github.com/chengcli/paddle"
-Repository = "https://github.com/chengcli/paddle"
-Issues = "https://github.com/chengcli/paddle/issues"
+Homepage = "https://github.com/elijah-mullens/paddle"
+Repository = "https://github.com/elijah-mullens/paddle"
+Issues = "https://github.com/elijah-mullens/paddle/issues"
 
 [project.scripts]
 paddle = "paddle.__main__:main"

{paddle-1.1.1 → paddle-1.1.3}/src/paddle/__init__.py

@@ -3,8 +3,5 @@ from .write_profile import write_profile
 from .find_init_params import find_init_params
 from .evolve_kinetics import evolve_kinetics
 
-__all__ = ["setup_profile",
-           "write_profile",
-           "find_init_params",
-           "evolve_kinetics"]
-__version__ = "1.1.1"
+__all__ = ["setup_profile", "write_profile", "find_init_params", "evolve_kinetics"]
+__version__ = "1.1.3"

{paddle-1.1.1 → paddle-1.1.3}/src/paddle/crm.py

@@ -4,20 +4,20 @@ import time
 import kintera
 import numpy as np
 from snapy import (
-        index,
-        MeshBlockOptions,
-        MeshBlock,
-        OutputOptions,
-        NetcdfOutput,
-        )
+    index,
+    MeshBlockOptions,
+    MeshBlock,
+    OutputOptions,
+    NetcdfOutput,
+)
 from kintera import (
-        ThermoOptions,
-        ThermoX,
-        KineticsOptions,
-        Kinetics,
-        )
+    ThermoOptions,
+    ThermoX,
+    KineticsOptions,
+    Kinetics,
+)
 
-if __name__ == '__main__':
+if __name__ == "__main__":
     # input file
     infile = "earth.yaml"
     device = "cpu"
@@ -46,7 +46,7 @@ if __name__ == '__main__':
 
     # set up initial condition
     w = setup_initial_condition(block, thermo_x)
-    print("w = ", w[:,0,0,:])
+    print("w = ", w[:, 0, 0, :])
 
     # integration
     current_time = 0.0
@@ -70,7 +70,7 @@ if __name__ == '__main__':
             block.forward(dt, stage)
 
         # evolve kinetics
-        u[index.icy:] += evolve_kinetics(block, kinet, thermo_x)
+        u[index.icy :] += evolve_kinetics(block, kinet, thermo_x)
 
         current_time += dt
         count += 1

{paddle-1.1.1 → paddle-1.1.3}/src/paddle/evolve_kinetics.py

@@ -2,12 +2,14 @@ import torch
 import snapy
 import kintera
 
+
 def evolve_kinetics(
     hydro_w: torch.Tensor,
-    block: snapy.MeshBlock, 
-    kinet: kintera.Kinetics, 
+    block: snapy.MeshBlock,
+    kinet: kintera.Kinetics,
     thermo_x: kintera.ThermoX,
-    dt) -> torch.Tensor:
+    dt,
+) -> torch.Tensor:
     """
     Evolve the chemical kinetics for one time step using implicit method.
 
@@ -26,7 +28,7 @@ def evolve_kinetics(
 
     temp = eos.compute("W->T", (hydro_w,))
     pres = hydro_w[snapy.index.ipr]
-    xfrac = thermo_y.compute("Y->X", (hydro_w[snapy.index.icy:],))
+    xfrac = thermo_y.compute("Y->X", (hydro_w[snapy.index.icy :],))
     conc = thermo_x.compute("TPX->V", (temp, pres, xfrac))
     cp_vol = thermo_x.compute("TV->cp", (temp, conc))
 

{paddle-1.1.1 → paddle-1.1.3}/src/paddle/find_init_params.py

@@ -4,16 +4,18 @@ import numpy as np
 
 from .setup_profile import setup_profile
 
+
 def find_init_params(
-        block: snapy.MeshBlock,
-        param: dict[str, float],
-        *,
-        target_T: float=300.,
-        target_P: float=1.e5, 
-        method: str="moist-adiabat",
-        max_iter: int=50,
-        ftol: float=1.e-2,
-        verbose: bool=True):
+    block: snapy.MeshBlock,
+    param: dict[str, float],
+    *,
+    target_T: float = 300.0,
+    target_P: float = 1.0e5,
+    method: str = "moist-adiabat",
+    max_iter: int = 50,
+    ftol: float = 1.0e-2,
+    verbose: bool = True,
+):
     """Find initial parameters that yield desired T and P
 
     Args:
@@ -44,14 +46,15 @@ def find_init_params(
         temp = eos.compute("W->T", (w,)).squeeze()
 
         # calculate 1D pressure
-        pres = w[snapy.index.ipr,...].squeeze()
+        pres = w[snapy.index.ipr, ...].squeeze()
 
         # temperature function
         t_func = interp1d(
             pres.log().cpu().numpy(),
             temp.log().cpu().numpy(),
             kind="linear",
-            fill_value="extrapolate")
+            fill_value="extrapolate",
+        )
 
         temp1 = np.exp(t_func(np.log(target_P)))
         if verbose:
@@ -69,4 +72,3 @@ def find_init_params(
         count += 1
 
     raise RuntimeError("Failed to converge within the maximum number of iterations.")
-

{paddle-1.1.1 → paddle-1.1.3}/src/paddle/setup_profile.py

@@ -4,18 +4,19 @@ import torch
 import snapy
 import kintera
 
+
 def integrate_neutral(
-        thermo_x: kintera.ThermoX,
-        temp: torch.Tensor,
-        pres: torch.Tensor,
-        xfrac: torch.Tensor,
-        grav: float,
-        dz: float,
-        max_iter: int = 100
-        ) -> Tuple[torch.Tensor, torch.Tensor, torch.Tensor]:
+    thermo_x: kintera.ThermoX,
+    temp: torch.Tensor,
+    pres: torch.Tensor,
+    xfrac: torch.Tensor,
+    grav: float,
+    dz: float,
+    max_iter: int = 100,
+) -> Tuple[torch.Tensor, torch.Tensor, torch.Tensor]:
     """
     A neutral density profile assumes no cloud and:
-        
+
         (1) dP/dz = -rho*g
         (2) d(rho)/dz = ...
 
@@ -68,15 +69,16 @@ def integrate_neutral(
 
     return temp2, pres2, xfrac2
 
+
 def integrate_dry_adiabat(
-        thermo_x: kintera.ThermoX,
-        temp: torch.Tensor,
-        pres: torch.Tensor,
-        xfrac: torch.Tensor,
-        grav: float,
-        dz: float,
-        max_iter: int = 100
-        ) -> Tuple[torch.Tensor, torch.Tensor, torch.Tensor]:
+    thermo_x: kintera.ThermoX,
+    temp: torch.Tensor,
+    pres: torch.Tensor,
+    xfrac: torch.Tensor,
+    grav: float,
+    dz: float,
+    max_iter: int = 100,
+) -> Tuple[torch.Tensor, torch.Tensor, torch.Tensor]:
     """
     A dry adiabatic profile assumes no cloud and:
 
@@ -84,12 +86,12 @@ def integrate_dry_adiabat(
         (2) dP/dz = -rho*g
 
     In discretized form:
-    
+
         cp_bar = 0.5 * (cp(T_old) + cp(T_new))
         T_new = T_old - g/bar * dz
         rho_bar = 0.5 * (rho_old + rho_new)
         P_new = P_old - rho_bar * g * dz
-        
+
     """
     conc1 = thermo_x.compute("TPX->V", [temp, pres, xfrac])
     cp1 = thermo_x.compute("TV->cp", [temp, conc1]) / conc1.sum(-1)
@@ -136,11 +138,10 @@ def integrate_dry_adiabat(
 
     return temp2, pres2, xfrac2
 
+
 def setup_profile(
-        block: snapy.MeshBlock,
-        param: dict[str, float] = {},
-        method: str = "moist-adiabat"
-        ) -> torch.Tensor:
+    block: snapy.MeshBlock, param: dict[str, float] = {}, method: str = "moist-adiabat"
+) -> torch.Tensor:
     """
     Set up an adiabatic initial condition for the mesh block.
 
@@ -175,16 +176,16 @@ def setup_profile(
         "moist-adiabat",
         "isothermal",
         "pseudo-adiabat",
-        "neutral"
+        "neutral",
     ]
 
     if method not in valid_methods:
         raise ValueError(f"Invalid method '{method}'. Choose from {valid_methods}.")
 
-    Ts = param.get("Ts", 300.)
-    Ps = param.get("Ps", 1.e5)
+    Ts = param.get("Ts", 300.0)
+    Ps = param.get("Ps", 1.0e5)
     grav = param.get("grav", 9.8)
-    Tmin = param.get("Tmin", 0.)
+    Tmin = param.get("Tmin", 0.0)
 
     # get handles to modules
     coord = block.module("hydro.coord")
@@ -204,8 +205,7 @@ def setup_profile(
     ny = len(thermo_y.options.species()) - 1
     nvar = 5 + ny
 
-    w = torch.zeros((nvar, nc3, nc2, nc1),
-                    dtype=x1v.dtype, device=x1v.device)
+    w = torch.zeros((nvar, nc3, nc2, nc1), dtype=x1v.dtype, device=x1v.device)
 
     temp = Ts * torch.ones((nc3, nc2), dtype=w.dtype, device=w.device)
     pres = Ps * torch.ones((nc3, nc2), dtype=w.dtype, device=w.device)
@@ -216,7 +216,7 @@ def setup_profile(
         xfrac[..., index] = param.get(f"x{name}", 0.0)
 
     # dry air mole fraction
-    xfrac[..., 0] = 1. - xfrac[..., 1:].sum(dim=-1)
+    xfrac[..., 0] = 1.0 - xfrac[..., 1:].sum(dim=-1)
 
     # start and end indices for the vertical direction
     # excluding ghost cells
@@ -227,7 +227,7 @@ def setup_profile(
     dz = coord.buffer("dx1f")[ifirst]
 
     # half a grid to cell center
-    thermo_x.extrapolate_ad(temp, pres, xfrac, grav, dz / 2.);
+    thermo_x.extrapolate_ad(temp, pres, xfrac, grav, dz / 2.0)
 
     # adiabatic extrapolation
     if method == "isothermal":
@@ -245,17 +245,19 @@ def setup_profile(
             xfrac /= xfrac.sum(dim=-1, keepdim=True)
         conc = thermo_x.compute("TPX->V", [temp, pres, xfrac])
 
-        w[snapy.index.ipr, ..., i] = pres;
+        w[snapy.index.ipr, ..., i] = pres
         w[snapy.index.idn, ..., i] = thermo_x.compute("V->D", [conc])
-        w[snapy.index.icy:, ...,i] = thermo_x.compute("X->Y", [xfrac])
+        w[snapy.index.icy :, ..., i] = thermo_x.compute("X->Y", [xfrac])
 
         dz = coord.buffer("dx1f")[i]
         if method.split("-")[0] == "dry":
-            temp, pres, xfrac = integrate_dry_adiabat(thermo_x, temp, pres, xfrac, grav, dz);
+            temp, pres, xfrac = integrate_dry_adiabat(
+                thermo_x, temp, pres, xfrac, grav, dz
+            )
         elif method.split("-")[0] == "neutral":
-            temp, pres, xfrac = integrate_neutral(thermo_x, temp, pres, xfrac, grav, dz);
+            temp, pres, xfrac = integrate_neutral(thermo_x, temp, pres, xfrac, grav, dz)
         else:
-            thermo_x.extrapolate_ad(temp, pres, xfrac, grav, dz);
+            thermo_x.extrapolate_ad(temp, pres, xfrac, grav, dz)
 
         if torch.any(temp < Tmin):
             i_isothermal = i + 1
@@ -276,5 +278,5 @@ def setup_profile(
         conc = thermo_x.compute("TPX->V", [temp, pres, xfrac])
         w[snapy.index.ipr, ..., i] = pres
         w[snapy.index.idn, ..., i] = thermo_x.compute("V->D", [conc])
-        w[snapy.index.icy:, ..., i] = thermo_x.compute("X->Y", [xfrac])
+        w[snapy.index.icy :, ..., i] = thermo_x.compute("X->Y", [xfrac])
     return w

{paddle-1.1.1 → paddle-1.1.3}/src/paddle/write_profile.py

@@ -7,11 +7,12 @@ import torch
 import kintera
 import snapy
 
+
 def write_profile(
     filename: str,
     hydro_w: torch.Tensor,
     block: snapy.MeshBlock,
-    ref_pressure: float = 1.e5,
+    ref_pressure: float = 1.0e5,
     comment: Optional[str] = None,
 ) -> None:
     """
@@ -46,27 +47,28 @@ def write_profile(
         raise ValueError("hydro_w must have shape (N, 1, 1, L).")
 
     # calculate a height grid
-    pres = hydro_w[snapy.index.ipr,...].squeeze() / 1.e5  # Pa -> bar
+    pres = hydro_w[snapy.index.ipr, ...].squeeze() / 1.0e5  # Pa -> bar
     zlev_func = interp1d(
         pres.log().cpu().numpy(),
         coord.buffer("x1v").cpu().numpy(),
         kind="linear",
-        fill_value="extrapolate")
-    zref = zlev_func(np.log(ref_pressure / 1.e5))
-    zlev = (coord.buffer("x1v") - zref) / 1.e3  # m -> km
+        fill_value="extrapolate",
+    )
+    zref = zlev_func(np.log(ref_pressure / 1.0e5))
+    zlev = (coord.buffer("x1v") - zref) / 1.0e3  # m -> km
 
     # calculate temperature
     temp = eos.compute("W->T", (hydro_w,)).squeeze()
 
     # calculate mole fractions
-    xfrac = thermo_y.compute("Y->X", (hydro_w[snapy.index.icy:,...],)).squeeze()
+    xfrac = thermo_y.compute("Y->X", (hydro_w[snapy.index.icy :, ...],)).squeeze()
 
     # calculate heat capacity
-    conc = thermo_x.compute("TPX->V", (temp, pres * 1.e5, xfrac))
+    conc = thermo_x.compute("TPX->V", (temp, pres * 1.0e5, xfrac))
     cpx = thermo_x.compute("TV->cp", (temp, conc)) / conc.sum(-1)
 
     # calculate entropy
-    ens = thermo_x.compute("TPV->S", (temp, pres * 1.e5, conc)) / conc.sum(-1)
+    ens = thermo_x.compute("TPV->S", (temp, pres * 1.0e5, conc)) / conc.sum(-1)
 
     with open(filename, "w") as f:
         # write comments

{paddle-1.1.1/src/paddle → paddle-1.1.3/tests}/data/saturn1d.yaml

@@ -1,5 +1,5 @@
 # Saturn Reference Atmosphere Model
-# 
+#
 # Solar abundances relative to H2, enrichments
 #
 # X_He  = 0.195, 0.6955

{paddle-1.1.1 → paddle-1.1.3}/tests/test_saturn_adiabat.py

@@ -1,45 +1,48 @@
 from importlib import resources
 from paddle import (
-        setup_profile,
-        write_profile,
-        find_init_params,
-        )
+    setup_profile,
+    write_profile,
+    find_init_params,
+)
 from snapy import (
-        MeshBlockOptions,
-        MeshBlock,
-        )
+    MeshBlockOptions,
+    MeshBlock,
+)
 from kintera import ThermoX
 
+
 def setup_saturn_profile():
-    path = resources.files("paddle") / "data" / "saturn1d.yaml"
+    # path = resources.files("paddle") / "data" / "saturn1d.yaml"
+    path = "data" / "saturn1d.yaml"
     print(f"Reading input file: {path}")
 
     op_block = MeshBlockOptions.from_yaml(str(path))
     block = MeshBlock(op_block)
 
     param = {
-        "Ts": 600.,
-        "Ps": 100.e5,
-        "Tmin": 85.,
+        "Ts": 600.0,
+        "Ps": 100.0e5,
+        "Tmin": 85.0,
         "xH2O": 8.91e-3,
         "xNH3": 3.52e-4,
         "xH2S": 8.08e-5,
         "grav": 10.44,
     }
 
-    #method = "pseudo-adiabat"
-    #method = "moist-adiabat"
+    # method = "pseudo-adiabat"
+    # method = "moist-adiabat"
     method = "dry-adiabat"
 
     param = find_init_params(
-            block,
-            param,
-            target_T=134.,
-            target_P=1.e5,
-            method=method,
-            max_iter=50,
-            ftol=1.e-2,
-            verbose=True)
+        block,
+        param,
+        target_T=134.0,
+        target_P=1.0e5,
+        method=method,
+        max_iter=50,
+        ftol=1.0e-2,
+        verbose=True,
+    )
 
     w = setup_profile(block, param, method=method)
 
@@ -50,5 +53,6 @@ def setup_saturn_profile():
     write_profile("saturn_profile.txt", w, block)
     return w
 
+
 if __name__ == "__main__":
     w = setup_saturn_profile()