update client with for the roughness stage

rosepearson · Oct 10, 2023 · 7202d14 · 7202d14
1 parent bda5b78
commit 7202d14
Showing 1 changed file with 11 additions and 10 deletions.
diff --git a/src/geofabrics/processor.py b/src/geofabrics/processor.py
@@ -1156,16 +1156,6 @@ def run(self):
             ),
         )
 
-        # Load in LiDAR tiles
-        self.roughness_dem.add_lidar(
-            lidar_datasets_info=lidar_datasets_info,
-            lidar_classifications_to_keep=self.get_instruction_general(
-                "lidar_classifications_to_keep"
-            ),
-            chunk_size=self.get_processing_instructions("chunk_size"),
-            metadata=self.create_metadata(),
-        )  # Note must be called after all others if it is to be complete
-
         # Setup Dask cluster and client
         cluster_kwargs = {
             "n_workers": self.get_processing_instructions("number_of_cores"),
@@ -1177,6 +1167,17 @@ def run(self):
         with cluster, distributed.Client(cluster) as client:
             print("Dask client:", client)
             print("Dask dashboard:", client.dashboard_link)
+
+            # Load in LiDAR tiles
+            self.roughness_dem.add_lidar(
+                lidar_datasets_info=lidar_datasets_info,
+                lidar_classifications_to_keep=self.get_instruction_general(
+                    "lidar_classifications_to_keep"
+                ),
+                chunk_size=self.get_processing_instructions("chunk_size"),
+                metadata=self.create_metadata(),
+            )  # Note must be called after all others if it is to be complete
+
             # save results
             try:
                 self.roughness_dem.dem.to_netcdf(