Skip to content

Commit

Permalink
Fix version of pandas package;
Browse files Browse the repository at this point in the history
Fix typing for tuple;
Remove untyped definition in first row;

PWPA-1930
  • Loading branch information
Gabriel Antão committed Jun 14, 2024
1 parent 2dca626 commit 9499349
Show file tree
Hide file tree
Showing 3 changed files with 3 additions and 4 deletions.
2 changes: 1 addition & 1 deletion setup.py
Original file line number Diff line number Diff line change
Expand Up @@ -11,7 +11,7 @@
"alfasim-sdk==0.20.0",
"attrs>=18.1.0",
"numpy>=1.11.0",
"pandas>=2.2.2",
"pandas>=2.0.0",
"oop-ext>=1.1",
"typing_extensions",
]
Expand Down
1 change: 0 additions & 1 deletion src/alfasim_score/converter/alfacase/convert_alfacase.py
Original file line number Diff line number Diff line change
@@ -1,4 +1,3 @@
# mypy: disallow-untyped-defs
from alfasim_sdk import AnnulusDescription
from alfasim_sdk import FormationDescription
from alfasim_sdk import ProfileDescription
Expand Down
4 changes: 2 additions & 2 deletions src/alfasim_score/converter/alfacase/score_input_reader.py
Original file line number Diff line number Diff line change
@@ -1,7 +1,7 @@
# mypy: disallow-untyped-defs
import json
from barril.units import Array
from pathlib import Path
from typing import Tuple

from alfasim_score.units import LENGTH_UNIT

Expand All @@ -12,7 +12,7 @@ def __init__(self, score_filepath: Path):
with open(score_filepath) as f:
self.input_content = json.load(f)

def read_well_trajectory(self) -> tuple[Array, Array]:
def read_well_trajectory(self) -> Tuple[Array, Array]:
"""Create the arrays with the x and y positions."""
x = [entry["displacement"] for entry in self.input_content["trajectory"]["data"]]
y = [-entry["vertical_depth"] for entry in self.input_content["trajectory"]["data"]]
Expand Down

0 comments on commit 9499349

Please sign in to comment.