Skip to content

Commit b15471d

Browse files
committed
changes as passed on my MJL
1 parent 22dfd63 commit b15471d

File tree

1 file changed

+10
-7
lines changed

1 file changed

+10
-7
lines changed

lineapy/graph_reader/types.py

Lines changed: 10 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -6,20 +6,23 @@ class InputVariable:
66
"""
77
Class to generate code related input variable and it's default value
88
9-
default_args: a = 1
10-
parser_body: parser.add_arguemnt('--a', default=1, type=)
11-
parser_args: a = args.a
9+
:param variable_name: variable name
10+
:param value: variable value
11+
:param value_type: variable objec type
12+
13+
Attributes:
14+
default_args: ex: ``a = 1``
15+
parser_body: ex: ``parser.add_arguemnt('--a', default=1, type=int)``
16+
parser_args: ex: ``a = args.a``
1217
1318
"""
1419

15-
def __init__(
16-
self, variable_name, value, value_type, typing_info=None
17-
) -> None:
20+
def __init__(self, variable_name, value, value_type) -> None:
1821
self.variable_name = variable_name
1922
self.value = value
2023
self.value_type = value_type.__name__
2124
# placeholder for future support to allow user passing typing info
22-
self.typing_info = typing_info
25+
# self.typing_info = typing_info
2326
self.default_args = f"{self.variable_name} = {repr(self.value)}"
2427
self.parser_body = f"parser.add_argument('--{self.variable_name}', type={self.value_type}, default={repr(self.value)})"
2528
self.parser_args = f"{self.variable_name} = args.{self.variable_name}"

0 commit comments

Comments
 (0)