Merge branch 'main' into v0.2.x

Author: lionsardesai

docs/source/autogen/lineapy.graph_reader.rst

@@ -34,6 +34,12 @@ lineapy.graph\_reader.session\_artifacts module
 .. automodule:: lineapy.graph_reader.session_artifacts
    :members:
 
+lineapy.graph\_reader.types module
+----------------------------------
+
+.. automodule:: lineapy.graph_reader.types
+   :members:
+
 lineapy.graph\_reader.utils module
 ----------------------------------
 

docs/source/autogen/lineapy.utils.rst

@@ -72,6 +72,12 @@ lineapy.utils.validate\_annotation\_spec module
 .. automodule:: lineapy.utils.validate_annotation_spec
    :members:
 
+lineapy.utils.version module
+----------------------------
+
+.. automodule:: lineapy.utils.version
+   :members:
+
 Module contents
 ---------------
 

lineapy/data/types.py

@@ -39,10 +39,9 @@ class SessionType(Enum):
 class SessionContext(BaseModel):
     """
     Each trace of a script/notebook is a "Session".
+
     :param working_directory: captures where the code ran by the user
 
-    - we should remove the dependency on the working_directory because
-      its brittle
     """
 
     id: LineaID  # populated on creation by uuid.uuid4()
@@ -87,11 +86,12 @@ class ValueType(Enum):
 
     TODO
     ----
-    FIXME
 
-    - rename (need coordination with linea-server):
+    rename (need coordination with linea-server)
+
         - really `dataset` is a table
         - `value` means its a literal  (e.g., int/str)
+
     """
 
     chart = 1
@@ -312,8 +312,11 @@ class ImportNode(BaseNode):
 
     These are optional because the info is acquired at runtime.
 
-    Note that this node is not actually used for execution (using `l_import` CallNodes),
-      but more a decoration for metadata retrieval.
+    .. note::
+
+        This node is not actually used for execution (using `l_import` CallNodes),
+        but more a decoration for metadata retrieval.
+
     """
 
     node_type: NodeType = NodeType.ImportNode

lineapy/db/db.py

@@ -594,8 +594,6 @@ def get_libraries_for_session(
         Gets all dependencies for a session, assuming all the libs in a
         particular session will be required to set up a new env.
 
-        TODO: I think this distinct is still broken, because we want to
-        make it distinct on a subset of columns: session_id, name, and version.
         """
         return (
             self.session.query(
@@ -682,7 +680,7 @@ def get_all_artifacts(self) -> List[ArtifactORM]:
     def get_nodes_for_session(self, session_id: LineaID) -> List[Node]:
         """
         Get all the nodes associated with the session, which does
-         NOT include things like SessionContext
+        NOT include things like SessionContext
         """
         node_orms = (
             self.session.query(BaseNodeORM)

lineapy/graph_reader/types.py

@@ -6,20 +6,23 @@ class InputVariable:
     """
     Class to generate code related input variable and it's default value
 
-    default_args: a = 1
-    parser_body: parser.add_arguemnt('--a', default=1, type=)
-    parser_args: a = args.a
+    :param variable_name: variable name
+    :param value: variable value
+    :param value_type: variable objec type
+
+    Attributes:
+        default_args: ex: ``a = 1``
+        parser_body: ex: ``parser.add_arguemnt('--a', default=1, type=int)``
+        parser_args: ex: ``a = args.a``
 
     """
 
-    def __init__(
-        self, variable_name, value, value_type, typing_info=None
-    ) -> None:
+    def __init__(self, variable_name, value, value_type) -> None:
         self.variable_name = variable_name
         self.value = value
         self.value_type = value_type.__name__
         # placeholder for future support to allow user passing typing info
-        self.typing_info = typing_info
+        # self.typing_info = typing_info
         self.default_args = f"{self.variable_name} = {repr(self.value)}"
         self.parser_body = f"parser.add_argument('--{self.variable_name}', type={self.value_type}, default={repr(self.value)})"
         self.parser_args = f"{self.variable_name} = args.{self.variable_name}"