move session down to the end of the class for diff

Author: zprobst

nodestream_plugin_neo4j/neo4j_database.py

@@ -84,9 +84,6 @@ def driver(self):
             self.acquire_driver()
         return self._driver
 
-    def session(self) -> AsyncSession:
-        return self.driver.session(database=self.database_name)
-
     def log_query_start(self, query: Query):
         self.logger.info(
             "Executing Cypher Query to Neo4j",
@@ -142,3 +139,6 @@ async def execute(
                 self.acquire_driver()
                 if attempts >= self.max_retry_attempts:
                     raise e
+
+    def session(self) -> AsyncSession:
+        return self.driver.session(database=self.database_name)