423A35C7/pytorch3d
1
0
Fork 0
mirror of https://github.com/facebookresearch/pytorch3d.git synced 2026-10-28 17:10:03 +08:00
Code Issues Packages Projects Releases Wiki Activity

Adding SQL Dataset related files to the build script

Browse Source 
Summary: Now that we have SQLAlchemy 2.0, we can fully use them.

Reviewed By: bottler

Differential Revision: D66920096

fbshipit-source-id: 25c0ea1c4f7361e66348035519627dc961b9e6e6
This commit is contained in:
Roman Shapovalov
2024-12-23 16:05:26 -08:00
committed by Facebook GitHub Bot
parent 055ab3a2e3
commit 64a5bfadc8
2 changed files with 56 additions and 34 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
pytorch3d/implicitron/dataset/sql_dataset_provider.py
View File

@@ -43,7 +43,7 @@ logger = logging.getLogger(__name__)
@registry.register
class SqlIndexDatasetMapProvider(DatasetMapProviderBase): # pyre-ignore [13]
class SqlIndexDatasetMapProvider(DatasetMapProviderBase):
"""
Generates the training, validation, and testing dataset objects for
a dataset laid out on disk like SQL-CO3D, with annotations in an SQLite data base.
@@ -193,9 +193,9 @@ class SqlIndexDatasetMapProvider(DatasetMapProviderBase): # pyre-ignore [13]
# this is a mould that is never constructed, used to build self._dataset_map values
dataset_class_type: str = "SqlIndexDataset"
dataset: SqlIndexDataset
dataset: SqlIndexDataset # pyre-ignore [13]
path_manager_factory: PathManagerFactory
path_manager_factory: PathManagerFactory # pyre-ignore [13]
path_manager_factory_class_type: str = "PathManagerFactory"
def __post_init__(self):