Skip to content

Commit

Permalink
Formatting
Browse files Browse the repository at this point in the history
  • Loading branch information
Martin Schubert authored and Martin Schubert committed Oct 2, 2023
1 parent 0b611d7 commit 2d439ca
Show file tree
Hide file tree
Showing 3 changed files with 5 additions and 7 deletions.
2 changes: 1 addition & 1 deletion README.md
Original file line number Diff line number Diff line change
Expand Up @@ -4,4 +4,4 @@
Some tests are marked as slow and are skipped by default. To run these manually, use
```
pytest --runslow
```
```
5 changes: 2 additions & 3 deletions tests/challenge/diffract/test_metagrating_challenge.py
Original file line number Diff line number Diff line change
@@ -1,18 +1,17 @@
"""Tests for `diffract.metagrating_challenge`."""

import dataclasses
import unittest

import dataclasses
import jax
import jax.numpy as jnp
import optax
from fmmax import fmm
from parameterized import parameterized
from totypes import symmetry # type: ignore[import,attr-defined,unused-ignore]
from fmmax import fmm

from invrs_gym.challenge.diffract import metagrating_challenge


LIGHTWEIGHT_SIM_PARAMS = dataclasses.replace(
metagrating_challenge.METAGRATING_SIM_PARAMS,
approximate_num_terms=100,
Expand Down
5 changes: 2 additions & 3 deletions tests/challenge/diffract/test_splitter_challenge.py
Original file line number Diff line number Diff line change
@@ -1,17 +1,16 @@
"""Tests for `diffract.splitter_challenge`."""

import dataclasses
import unittest

import dataclasses
import jax
import jax.numpy as jnp
import optax
from parameterized import parameterized
from fmmax import fmm
from parameterized import parameterized

from invrs_gym.challenge.diffract import splitter_challenge


LIGHTWEIGHT_SIM_PARAMS = dataclasses.replace(
splitter_challenge.DIFFRACTIVE_SPLITTER_SIM_PARAMS,
approximate_num_terms=100,
Expand Down

0 comments on commit 2d439ca

Please sign in to comment.