feedback 4

Author: robertMileaNi

nisystemlink/clients/notebook/_notebook_client.py

@@ -256,7 +256,7 @@ def get_execution_by_id(self, id: str) -> models.Execution:
     @post("ninbexecution/v1/query-executions")
     def __query_executions(
         self, query: models._QueryExecutionsRequest
-    ) -> List[models.QueryExecutionResponse]:
+    ) -> List[models.Execution]:
         """Query executions of Jupyter notebooks.
 
         Args:
@@ -273,7 +273,7 @@ def __query_executions(
 
     def query_executions(
         self, query: models.QueryExecutionsRequest
-    ) -> List[models.QueryExecutionResponse]:
+    ) -> List[models.Execution]:
         """Query executions of Jupyter notebooks.
 
         Args:

nisystemlink/clients/notebook/models/__init__.py

@@ -7,9 +7,6 @@
     _QueryExecutionsRequest,
     QueryExecutionsRequest,
 )
-from ._query_execution_response import (
-    QueryExecutionResponse,
-)
 from ._create_execution_request import (
     CreateExecutionRequest,
 )