bug fix

Author: yashlimbad

dataframe_test.py

@@ -30,11 +30,10 @@ def tearDown(self) -> None:
         self.disconnect()
 
     def test_table_creation(self):
-        self._dataframe = self.e6x_connection.load_parquet('<file_path>')
-        self._dataframe.select('col1','col2')
-
-        rows = self._dataframe.show()
-
-        for row in rows:
-            print(row)
-
+        try:
+            self._dataframe = self.e6x_connection.load_parquet('<filepath>')
+            rows = self._dataframe.show()
+            for row in rows:
+                print(row)
+        except BaseException as e:
+            print(f"Exception :{e}")

e6data_python_connector/e6data_grpc.py

@@ -1275,7 +1275,7 @@ def __exit__(self, exc_type, exc_val, exc_tb):
         self.terminate()
 
     def update_dataframe_map(self, dataframe : "DataFrame"):
-        self._dataframe_map.update({self._dataframe_count, dataframe})
+        self._dataframe_map.update({self._dataframe_count : dataframe})
         self._dataframe_count = self._dataframe_count + 1
 
     @property