Skip to content

Commit

Permalink
docs: add docstrings to the input module
Browse files Browse the repository at this point in the history
  • Loading branch information
rabii-chaarani committed Jul 10, 2024
1 parent e412abe commit 4f73437
Showing 1 changed file with 17 additions and 0 deletions.
17 changes: 17 additions & 0 deletions LoopDataConverter/input/input_data.py
Original file line number Diff line number Diff line change
Expand Up @@ -5,13 +5,30 @@

@dataclass
class InputData:
""" Class to store input data for the loop data converter
Attributes:
geology: Datatype.GEOLOGY = None
structure: Datatype.STRUCTURE = None
fault: Datatype.FAULT = None
fold: Datatype.FOLD = None
"""
geology: Datatype.GEOLOGY = None
structure: Datatype.STRUCTURE = None
fault: Datatype.FAULT = None
fold: Datatype.FOLD = None


def __getitem__(self, datatype: Datatype):
""" Method to get the the file directory of a datatype
Parameters:
datatype (Datatype): The datatype to get the file directory of
Returns:
The file directory of the datatype
"""

return self.__dict__[datatype]

Expand Down

0 comments on commit 4f73437

Please sign in to comment.