Skip to content

Commit

Permalink
Fixed linting
Browse files Browse the repository at this point in the history
  • Loading branch information
marcosertoli committed Jul 17, 2024
1 parent 67c6719 commit 2d821ee
Show file tree
Hide file tree
Showing 2 changed files with 12 additions and 5 deletions.
10 changes: 6 additions & 4 deletions indica/examples/example_plasma.py
Original file line number Diff line number Diff line change
@@ -1,9 +1,12 @@
from pathlib import Path
import pickle
from typing import Tuple

import numpy as np

from indica.models import Plasma
from indica.models.plasma import PlasmaProfiles
import numpy as np
from typing import Tuple
from pathlib import Path


def example_plasma(
machine: str = "st40",
Expand Down Expand Up @@ -80,4 +83,3 @@ def example_plasma(
pickle.dump(plasma, open(default_plasma_file, "wb"))

return plasma

7 changes: 6 additions & 1 deletion indica/examples/example_transforms.py
Original file line number Diff line number Diff line change
@@ -1,5 +1,8 @@
import numpy as np
from indica.converters import TransectCoordinates, LineOfSightTransform

from indica.converters import LineOfSightTransform
from indica.converters import TransectCoordinates


def pi_transform_example(nchannels: int):
x_positions = np.linspace(0.2, 0.8, nchannels)
Expand Down Expand Up @@ -39,6 +42,7 @@ def helike_transform_example(nchannels):
)
return los_transform


def smmh1_transform_example(nchannels):
los_start = np.array([[0.8, 0, 0]]) * np.ones((nchannels, 3))
los_start[:, 2] = np.linspace(0, -0.2, nchannels)
Expand All @@ -59,6 +63,7 @@ def smmh1_transform_example(nchannels):
)
return los_transform


def ts_transform_example(nchannels):
x_positions = np.linspace(0.2, 0.8, nchannels)
y_positions = np.linspace(0.0, 0.0, nchannels)
Expand Down

0 comments on commit 2d821ee

Please sign in to comment.