1717from distributed .client import FutureCancelledError
1818from distributed .scheduler import KilledWorker
1919from distributed .shuffle ._exceptions import P2PConsistencyError
20- from xarray .coding .times import CFDatetimeCoder
2120from xarray import MergeError
21+ from xarray .coding .times import CFDatetimeCoder
2222
2323from aodn_cloud_optimised .lib .CommonHandler import CommonHandler
2424from aodn_cloud_optimised .lib .logging import get_logger
@@ -1083,7 +1083,9 @@ def publish_cloud_optimised_fileset_batch(self, s3_file_uri_list):
10831083 if attrs .get ("drop_var" , False )
10841084 ]
10851085
1086- partial_preprocess = partial (preprocess_xarray , dataset_config = self .dataset_config )
1086+ partial_preprocess = partial (
1087+ preprocess_xarray , dataset_config = self .dataset_config
1088+ )
10871089
10881090 if self .cluster_mode :
10891091 batch_size = self .get_batch_size (client = self .client )
@@ -1800,12 +1802,14 @@ def handle_append_dim_overlap(
18001802 batch_files ,
18011803 dim_name = dim_name ,
18021804 dataset_config = self .dataset_config ,
1803- uuid_log = self .uuid_log
1805+ uuid_log = self .uuid_log ,
18041806 )
18051807 ranges = self .client .gather (futures )
18061808 else :
18071809 ranges = [
1808- check_append_dim_range_dask (f , dim_name , self .dataset_config , self .uuid_log )
1810+ check_append_dim_range_dask (
1811+ f , dim_name , self .dataset_config , self .uuid_log
1812+ )
18091813 for f in batch_files
18101814 ]
18111815
@@ -1957,7 +1961,7 @@ def check_variable_values_parallel(self, file_paths, variable_name):
19571961 reference_values = reference_values ,
19581962 variable_name = variable_name ,
19591963 dataset_config = self .dataset_config ,
1960- uuid_log = self .uuid_log
1964+ uuid_log = self .uuid_log ,
19611965 )
19621966 results = self .client .gather (futures )
19631967 else :
@@ -1972,7 +1976,7 @@ def check_variable_values_parallel(self, file_paths, variable_name):
19721976 reference_values ,
19731977 variable_name ,
19741978 self .dataset_config ,
1975- self .uuid_log
1979+ self .uuid_log ,
19761980 )
19771981 for file_path in file_paths [1 :]
19781982 ]
0 commit comments