Skip to content

Commit

Permalink
chore: fix linting issues
Browse files Browse the repository at this point in the history
  • Loading branch information
kjy5 committed Mar 27, 2024
1 parent c4a9931 commit 2ad6f93
Showing 1 changed file with 12 additions and 9 deletions.
21 changes: 12 additions & 9 deletions src/vbl_aquarium/models/urchin.py
Original file line number Diff line number Diff line change
Expand Up @@ -3,7 +3,6 @@
from enum import Enum

from pydantic import Field
from typing import List

from vbl_aquarium.models.unity import Color, Vector2, Vector3
from vbl_aquarium.utils.vbl_base_model import VBLBaseModel
Expand Down Expand Up @@ -42,8 +41,9 @@ class CustomMeshModel(VBLBaseModel):
class AtlasModel(VBLBaseModel):
name: str
reference_coord: Vector3 = None
areas: List[StructureModel]
colormap: str = ''
areas: list[StructureModel]
colormap: str = ""


class StructureModel(VBLBaseModel):
name: str
Expand All @@ -54,7 +54,8 @@ class StructureModel(VBLBaseModel):
visible: bool = False
color_intensity: float = -1
side: int = 0
material: str = 'default'
material: str = "default"


# Camera

Expand All @@ -78,8 +79,10 @@ class CameraMode(str, Enum):

# Individual mesh neuron


class PrimitiveMeshModel(VBLBaseModel):
data: List[MeshModel]
data: list[MeshModel]


class MeshModel(VBLBaseModel):
id: str = Field(alias="ID")
Expand All @@ -100,8 +103,8 @@ class ParticleGroupModel(VBLBaseModel):
shape: str
material: str

xs: List[float]
ys: List[float]
zs: List[float]
xs: list[float]
ys: list[float]
zs: list[float]

colors: List[Color]
colors: list[Color]

0 comments on commit 2ad6f93

Please sign in to comment.