[tfds] Fix pytype error

PiperOrigin-RevId: 339841828

Author: Conchylicultor

tensorflow_graphics/datasets/modelnet40/__init__.py

@@ -63,7 +63,7 @@ class ModelNet40(tfds.core.GeneratorBasedBuilder):
 
   @staticmethod
   def load(*args, **kwargs):
-    return tfds.load('model_net40', *args, **kwargs)
+    return tfds.load('model_net40', *args, **kwargs)  # pytype: disable=wrong-arg-count
 
   def _info(self):
     return tfds.core.DatasetInfo(

tensorflow_graphics/datasets/shapenet/shapenet.py

@@ -205,7 +205,7 @@ class Shapenet(tfds.core.GeneratorBasedBuilder):
 
   @staticmethod
   def load(*args, **kwargs):
-    return tfds.load('shapenet', *args, **kwargs)
+    return tfds.load('shapenet', *args, **kwargs)  # pytype: disable=wrong-arg-count
 
   MANUAL_DOWNLOAD_INSTRUCTIONS = textwrap.dedent("""\
   manual_dir should contain the extracted ShapeNetCore.v2.zip archive.