Removes dependency on singledispatchmethod package which is included …

…in functools since Python 3.8 (#317)

PiperOrigin-RevId: 619494372

Co-authored-by: kubric-team <kubric@google.com>

kubric/renderer/blender.py

@@ -15,27 +15,25 @@
 # pylint: disable=function-redefined (removes singledispatchmethod pylint errors)
 
 import collections
+from contextlib import redirect_stdout
+import functools
 import io
 import logging
 import os
 import sys
-from contextlib import redirect_stdout
-from typing import Any, Dict, Optional, Sequence, Union
 import tempfile
-
-from kubric.safeimport.bpy import bpy
-
-import numpy as np
-import tensorflow as tf
-from singledispatchmethod import singledispatchmethod
+from typing import Any, Dict, Optional, Sequence, Union
 
 import kubric as kb
 from kubric import core
+from kubric import file_io
 from kubric.core.assets import UndefinedAsset
+from kubric.file_io import PathLike
 from kubric.redirect_io import RedirectStream
 from kubric.renderer import blender_utils
-from kubric import file_io
-from kubric.file_io import PathLike
+from kubric.safeimport.bpy import bpy
+import numpy as np
+import tensorflow as tf
 
 logger = logging.getLogger(__name__)
 
@@ -358,7 +356,7 @@ def clear_and_reset_blender_scene(verbose: bool = False, custom_scene: str = Non
         logger.info("Loading scene from '%s'", custom_scene)
         bpy.ops.wm.open_mainfile(filepath=custom_scene)
 
-  @singledispatchmethod
+  @functools.singledispatchmethod
   def add_asset(self, asset: core.Asset) -> Any:
     raise NotImplementedError(f"Cannot add {asset!r}")
 

kubric/simulator/pybullet.py

@@ -22,11 +22,9 @@
 import tempfile
 from typing import Dict, List, Optional, Tuple, Union
 
-import tensorflow as tf
-from singledispatchmethod import singledispatchmethod
-
 from kubric import core
 from kubric.redirect_io import RedirectStream
+import tensorflow as tf
 
 # --- hides the "pybullet build time: May 26 2021 18:52:36" message on import
 with RedirectStream(stream=sys.stderr):
@@ -86,7 +84,7 @@ def __init__(self, scene: core.Scene, scratch_dir=tempfile.mkdtemp()):
   def physics_client(self):
     return self._physics_client.client
 
-  @singledispatchmethod
+  @functools.singledispatchmethod
   def add_asset(self, asset: core.Asset) -> Optional[int]:
     raise NotImplementedError(f"Cannot add {asset!r}")
 

requirements.txt

@@ -13,7 +13,6 @@ pypng
 pyquaternion
 python-Levenshtein
 scikit-learn
-singledispatchmethod
 tensorflow
 tensorflow-datasets>=4.1.0
 traitlets