diff --git a/docs/physics/state.rst b/docs/physics/state.rst index e56dfb07..e9d5e37a 100644 --- a/docs/physics/state.rst +++ b/docs/physics/state.rst @@ -35,7 +35,7 @@ You can initialize a zero-filled PhysicsState and MicrophysicsState from other P ... tile_rank=communicator.tile.rank, ... ) - >>> quantity_factory = QuantityFactory.from_backend(sizer=sizer, backend="numpy") + >>> quantity_factory = QuantityFactory(sizer=sizer, backend="numpy") >>> physics_state = PhysicsState.init_zeros( ... quantity_factory=quantity_factory, schemes=["GFS_microphysics"] ... ) diff --git a/examples/notebooks/functions.py b/examples/notebooks/functions.py index 0c81ff58..0c1356cc 100644 --- a/examples/notebooks/functions.py +++ b/examples/notebooks/functions.py @@ -268,7 +268,7 @@ def configure_domain( tile_rank=communicator.tile.rank, ) - quantity_factory = QuantityFactory.from_backend(sizer=sizer, backend=backend) + quantity_factory = QuantityFactory(sizer=sizer, backend=backend) metric_terms = MetricTerms( quantity_factory=quantity_factory, diff --git a/examples/notebooks/grid_generation.ipynb b/examples/notebooks/grid_generation.ipynb index eb331536..e7ab6441 100644 --- a/examples/notebooks/grid_generation.ipynb +++ b/examples/notebooks/grid_generation.ipynb @@ -188,7 +188,7 @@ ")\n", "\n", "# useful for easily allocating distributed data storages (fields)\n", - "quantity_factory = QuantityFactory.from_backend(sizer=sizer, backend=backend)" + "quantity_factory = QuantityFactory(sizer=sizer, backend=backend)" ] }, { diff --git a/pace/driver.py b/pace/driver.py index 4760ff58..8fa46442 100644 --- a/pace/driver.py +++ b/pace/driver.py @@ -177,7 +177,7 @@ def get_grid( tile_partitioner=communicator.partitioner.tile, tile_rank=communicator.tile.rank, ) - quantity_factory = QuantityFactory.from_backend( + quantity_factory = QuantityFactory( sizer, backend=self.stencil_config.compilation_config.backend ) @@ -207,7 +207,7 @@ def get_driver_state( tile_rank=communicator.tile.rank, ) if quantity_factory is None: - quantity_factory = QuantityFactory.from_backend( + quantity_factory = QuantityFactory( sizer, backend=self.stencil_config.compilation_config.backend ) if stencil_factory is None: @@ -760,7 +760,7 @@ def _setup_factories( grid_indexing = GridIndexing.from_sizer_and_communicator( sizer=sizer, comm=communicator ) - quantity_factory = QuantityFactory.from_backend( + quantity_factory = QuantityFactory( sizer, backend=config.stencil_config.compilation_config.backend ) stencil_factory = StencilFactory( diff --git a/pace/state.py b/pace/state.py index 9caa1dc6..d3d3b0dc 100644 --- a/pace/state.py +++ b/pace/state.py @@ -91,7 +91,7 @@ def load_state_from_restart( tile_partitioner=communicator.partitioner.tile, tile_rank=communicator.tile.rank, ) - quantity_factory = QuantityFactory.from_backend( + quantity_factory = QuantityFactory( sizer, backend=driver_config.stencil_config.compilation_config.backend ) diff --git a/tests/main/driver/test_diagnostics_config.py b/tests/main/driver/test_diagnostics_config.py index cd0b2ca0..a84a475b 100644 --- a/tests/main/driver/test_diagnostics_config.py +++ b/tests/main/driver/test_diagnostics_config.py @@ -42,7 +42,7 @@ def test_zselect_raises_error_if_not_3d(tmpdir): z_select=[ZSelect(level=0, names=["phis"])], ) result = config.diagnostics_factory(unittest.mock.MagicMock()) - quantity_factory = QuantityFactory.from_backend( + quantity_factory = QuantityFactory( sizer=SubtileGridSizer.from_tile_params( nx_tile=12, ny_tile=12, @@ -63,7 +63,7 @@ def test_zselect_raises_error_if_3rd_dim_not_z(tmpdir): z_select=[ZSelect(level=0, names=["foo"])], ) result = config.diagnostics_factory(unittest.mock.MagicMock()) - quantity_factory = QuantityFactory.from_backend( + quantity_factory = QuantityFactory( sizer=SubtileGridSizer.from_tile_params( nx_tile=12, ny_tile=12, diff --git a/tests/main/driver/test_restart_fortran.py b/tests/main/driver/test_restart_fortran.py index c658fc02..b355e206 100644 --- a/tests/main/driver/test_restart_fortran.py +++ b/tests/main/driver/test_restart_fortran.py @@ -35,7 +35,7 @@ def test_state_from_fortran_restart(): tile_rank=0, ) - quantity_factory = QuantityFactory.from_backend(sizer=sizer, backend="numpy") + quantity_factory = QuantityFactory(sizer=sizer, backend="numpy") restart_dir = REPO_ROOT / "tests" / "main" / "data" / "c12_restart" ( diff --git a/tests/main/driver/test_restart_serial.py b/tests/main/driver/test_restart_serial.py index aa9fe693..ddbed233 100644 --- a/tests/main/driver/test_restart_serial.py +++ b/tests/main/driver/test_restart_serial.py @@ -67,7 +67,7 @@ def test_restart_save_to_disk(): tile_partitioner=partitioner.tile, tile_rank=communicator.tile.rank, ) - quantity_factory = QuantityFactory.from_backend(sizer=sizer, backend=backend) + quantity_factory = QuantityFactory(sizer=sizer, backend=backend) eta_file = Path(driver_config.grid_config.config.eta_file) ( diff --git a/tests/main/fv3core/test_dycore_baroclinic.py b/tests/main/fv3core/test_dycore_baroclinic.py index afebd877..c3f214e1 100644 --- a/tests/main/fv3core/test_dycore_baroclinic.py +++ b/tests/main/fv3core/test_dycore_baroclinic.py @@ -112,7 +112,7 @@ def setup_dycore( grid_indexing = GridIndexing.from_sizer_and_communicator( sizer=sizer, comm=communicator ) - quantity_factory = QuantityFactory.from_backend(sizer=sizer, backend=backend) + quantity_factory = QuantityFactory(sizer=sizer, backend=backend) eta_file = "NDSL/tests/data/eta/eta79.nc" metric_terms = MetricTerms( quantity_factory=quantity_factory, diff --git a/tests/main/fv3core/test_dycore_call.py b/tests/main/fv3core/test_dycore_call.py index e41f6972..e797828f 100644 --- a/tests/main/fv3core/test_dycore_call.py +++ b/tests/main/fv3core/test_dycore_call.py @@ -96,7 +96,7 @@ def setup_dycore() -> Tuple[DynamicalCore, DycoreState, Timer]: grid_indexing = GridIndexing.from_sizer_and_communicator( sizer=sizer, comm=communicator ) - quantity_factory = QuantityFactory.from_backend(sizer=sizer, backend=backend) + quantity_factory = QuantityFactory(sizer=sizer, backend=backend) eta_file = Path("NDSL/tests/data/eta/eta79.nc") metric_terms = MetricTerms( quantity_factory=quantity_factory, diff --git a/tests/main/physics/test_integration.py b/tests/main/physics/test_integration.py index 6e215ea5..ff75d74f 100644 --- a/tests/main/physics/test_integration.py +++ b/tests/main/physics/test_integration.py @@ -50,7 +50,7 @@ def setup_physics(): grid_indexing = GridIndexing.from_sizer_and_communicator( sizer=sizer, comm=communicator ) - quantity_factory = QuantityFactory.from_backend(sizer=sizer, backend=backend) + quantity_factory = QuantityFactory(sizer=sizer, backend=backend) dace_config = DaceConfig( communicator=communicator, backend=backend, diff --git a/tests/main/test_grid_init.py b/tests/main/test_grid_init.py index 32f05aed..2c28a217 100644 --- a/tests/main/test_grid_init.py +++ b/tests/main/test_grid_init.py @@ -25,7 +25,7 @@ def get_cube_comm(layout, rank: int): def get_quantity_factory(layout, nx_tile, ny_tile, nz): nx = nx_tile // layout[0] ny = ny_tile // layout[1] - return QuantityFactory.from_backend( + return QuantityFactory( sizer=SubtileGridSizer.from_tile_params( nx_tile=nx, ny_tile=ny, diff --git a/tests/mpi/test_grid_init.py b/tests/mpi/test_grid_init.py index 7e0ab41a..e472ab88 100644 --- a/tests/mpi/test_grid_init.py +++ b/tests/mpi/test_grid_init.py @@ -27,7 +27,7 @@ def get_cube_comm(layout, comm: MPIComm): def get_quantity_factory(layout, nx_tile, ny_tile, nz): nx = nx_tile // layout[0] ny = ny_tile // layout[1] - return QuantityFactory.from_backend( + return QuantityFactory( sizer=SubtileGridSizer.from_tile_params( nx=nx, ny=ny, nz=nz, n_halo=3, layout=(1, 1) ), diff --git a/tests/savepoint/translate/translate_driver.py b/tests/savepoint/translate/translate_driver.py index f3499c56..231f2f92 100644 --- a/tests/savepoint/translate/translate_driver.py +++ b/tests/savepoint/translate/translate_driver.py @@ -36,7 +36,7 @@ def compute_parallel(self, inputs, communicator): tile_rank=communicator.tile.rank, ) - quantity_factory = QuantityFactory.from_backend( + quantity_factory = QuantityFactory( sizer, backend=self.stencil_config.compilation_config.backend ) physics_state = PhysicsState.init_zeros(