add jaxlib

tlambert03 · Apr 9, 2024 · ad95c11 · ad95c11
1 parent b221366
commit ad95c11
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -48,7 +48,7 @@ dependencies = [
 
 # https://peps.python.org/pep-0621/#dependencies-optional-dependencies
 [project.optional-dependencies]
-test = ["pytest", "pytest-cov", "zarr", "tifffile", "jax"]
+test = ["pytest", "pytest-cov", "zarr", "tifffile", "jax[cpu]"]
 dev = [
     "microsim[test]",
     "ipython",

diff --git a/src/microsim/schema/simulation.py b/src/microsim/schema/simulation.py
@@ -30,7 +30,9 @@ class Simulation(BaseModel):
     output_space: Space | None = None
     sample: Sample
     objective_lens: ObjectiveLens = Field(default_factory=ObjectiveLens)
-    channels: list[Channel]
+    channels: list[Channel] = Field(
+        default_factory=lambda: [Channel(name="FITC", excitation=488, emission=525)]
+    )
     modality: Modality = Field(default_factory=Widefield)
     settings: Settings = Field(default_factory=Settings)
     output: OutPath | None = None