From 2e42ef793f6be12627e820cf9878bd2da1602c80 Mon Sep 17 00:00:00 2001 From: Jeremy Reizenstein Date: Wed, 22 Jun 2022 04:24:14 -0700 Subject: [PATCH] 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 | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/pytorch3d/implicitron/dataset/data_source.py b/pytorch3d/implicitron/dataset/data_source.py index d8c1f805..0dae079d 100644 --- a/pytorch3d/implicitron/dataset/data_source.py +++ b/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