initialize state instead of content in list

Author: william-lukusa

dataikuapi/dss/notebook.py

@@ -41,6 +41,7 @@ def get_state(self):
         for notebook in notebook_list:
             if notebook.get("name") == self.notebook_name:
                     self.state = notebook
+                    return self.state
         return self.state
 
     def get_sessions(self):

dataikuapi/dss/project.py

@@ -844,7 +844,7 @@ def list_jupyter_notebooks(self, as_objects=True, active=False):
         notebooks = self.client._perform_json("GET", "/projects/%s/jupyter-notebooks/" % self.project_key,
                                          params={"active": active})
         if as_objects:
-            return [DSSNotebook(self.client, notebook['projectKey'], notebook['name'], content=notebook) for notebook in notebooks]
+            return [DSSNotebook(self.client, notebook_state['projectKey'], notebook_state['name'], state=notebook_state) for notebook_state in notebooks]
         else:
             return notebooks