register ImplicitronDataSource

Summary: Just register ImplicitronDataSource. We don't use it as pluggable yet here.

Reviewed By: shapovalov

Differential Revision: D37315698

fbshipit-source-id: ac41153383f9ab6b14ac69a3dfdc44aca0d94995

pytorch3d/implicitron/dataset/data_source.py

@@ -6,7 +6,11 @@
 
 from typing import Tuple
 
-from pytorch3d.implicitron.tools.config import ReplaceableBase, run_auto_creation
+from pytorch3d.implicitron.tools.config import (
+    registry,
+    ReplaceableBase,
+    run_auto_creation,
+)
 
 from .blender_dataset_map_provider import BlenderDatasetMapProvider  # noqa
 from .data_loader_map_provider import DataLoaderMap, DataLoaderMapProviderBase
@@ -25,6 +29,7 @@ class DataSourceBase(ReplaceableBase):
         raise NotImplementedError()
 
 
+@registry.register
 class ImplicitronDataSource(DataSourceBase):  # pyre-ignore[13]
     """
     Represents the data used in Implicitron. This is the only implementation