deploy: 32a5700 [formerly 7a32302ac1427cf87dcbc2d69001d66345a2fcdb]

Former-commit-id: 0364d2e2c5ecb08ba86709af48947f0509ab4ef1
FAIR-Chem · Dec 19, 2024 · fcbf3a5 · fcbf3a5
1 parent 96e55d5
commit fcbf3a5
Show file tree

Hide file tree

Showing 5 changed files with 6 additions and 3 deletions.
diff --git a/_sources/autoapi/core/common/relaxation/ase_utils/index.rst b/_sources/autoapi/core/common/relaxation/ase_utils/index.rst
@@ -59,7 +59,7 @@ Module Contents
    :returns: list of Atoms
 
 
-.. py:class:: OCPCalculator(config_yml: str | None = None, checkpoint_path: str | pathlib.Path | None = None, model_name: str | None = None, local_cache: str | None = None, trainer: str | None = None, cpu: bool = True, seed: int | None = None)
+.. py:class:: OCPCalculator(config_yml: str | None = None, checkpoint_path: str | pathlib.Path | None = None, model_name: str | None = None, local_cache: str | None = None, trainer: str | None = None, cpu: bool = True, seed: int | None = None, only_output: list[str] | None = None)
 
    Bases: :py:obj:`ase.calculators.calculator.Calculator`
 

diff --git a/_sources/autoapi/core/common/utils/index.rst b/_sources/autoapi/core/common/utils/index.rst
@@ -80,6 +80,7 @@ Functions
    core.common.utils.get_timestamp_uid
    core.common.utils.tensor_stats
    core.common.utils.get_weight_table
+   core.common.utils.get_checkpoint_format
 
 
 Module Contents
@@ -306,3 +307,5 @@ Module Contents
 
 .. py:function:: get_weight_table(model: torch.nn.Module) -> tuple[list, list]
 
+.. py:function:: get_checkpoint_format(config: dict) -> str
+
diff --git a/_sources/autoapi/core/trainers/base_trainer/index.rst b/_sources/autoapi/core/trainers/base_trainer/index.rst
@@ -127,7 +127,7 @@ Module Contents
    .. py:method:: get_sampler(dataset, batch_size: int, shuffle: bool) -> fairchem.core.common.data_parallel.BalancedBatchSampler
 
 
-   .. py:method:: get_dataloader(dataset, sampler) -> torch.utils.data.DataLoader
+   .. py:method:: get_dataloader(dataset, sampler, workers=None) -> torch.utils.data.DataLoader
 
 
    .. py:method:: load_datasets() -> None

diff --git a/_sources/autoapi/core/trainers/index.rst b/_sources/autoapi/core/trainers/index.rst
@@ -130,7 +130,7 @@ Package Contents
    .. py:method:: get_sampler(dataset, batch_size: int, shuffle: bool) -> fairchem.core.common.data_parallel.BalancedBatchSampler
 
 
-   .. py:method:: get_dataloader(dataset, sampler) -> torch.utils.data.DataLoader
+   .. py:method:: get_dataloader(dataset, sampler, workers=None) -> torch.utils.data.DataLoader
 
 
    .. py:method:: load_datasets() -> None

diff --git a/objects.inv b/objects.inv