style(utility): update error message for calling "get_callback_body"

PR Closed: #1246
Graviti-AI · Apr 11, 2022 · 9d5e031 · 9d5e031
1 parent 526a7e9
commit 9d5e031
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 1 deletion.
diff --git a/tensorbay/client/segment.py b/tensorbay/client/segment.py
@@ -933,7 +933,7 @@ def upload_frame(self, frame: Frame, timestamp: Optional[float] = None) -> None:
         callback_bodies = []
         for sensor_name, data in frame.items():
             try:
-                callback_body = data.get_callback_body()  # type:ignore[union-attr]
+                callback_body = data.get_callback_body()
             except AttributeError:
                 continue
 

diff --git a/tensorbay/utility/file.py b/tensorbay/utility/file.py
@@ -220,3 +220,12 @@ def open(self) -> Union[UserResponse, BufferedReader]:
             self._write_cache(cache_path)
 
         return open(cache_path, "rb")
+
+    def get_callback_body(self) -> Dict[str, Any]:
+        """Not support ``get_callback_body`` function.
+
+        Raises:
+            AttributeError: when calling the ``get_callback_body()``.
+
+        """
+        raise AttributeError(f"Could not dump a '{self.__class__.__name__}' instance")