[FIX] Rename method for label return

wbenbihi · Aug 23, 2022 · 9614623 · 9614623
1 parent 2daad12
commit 9614623
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/hourglass_tensorflow/handlers/data.py b/hourglass_tensorflow/handlers/data.py
@@ -181,7 +181,7 @@ def prepare_output(self, _error: bool = True) -> None:
         self._validate_joints(_error=_error)
         self._read_labels(_error=_error)
 
-    def get_return(self) -> pd.DataFrame:
+    def get_data(self) -> pd.DataFrame:
         return self._labels_df