File tree Expand file tree Collapse file tree 3 files changed +603
-509
lines changed
Expand file tree Collapse file tree 3 files changed +603
-509
lines changed Original file line number Diff line number Diff line change @@ -274,6 +274,7 @@ def wrapper(
274274 normalize_target_data : bool = True ,
275275 target_custom_scale_factor : dict = {},
276276 num_threads : int = None ,
277+ iteratively_update_sigma : bool = False ,
277278 ):
278279 if subset_data is None :
279280 raise ValueError ("Subset data cannot be None" )
@@ -306,6 +307,8 @@ def wrapper(
306307 if num_threads is not None :
307308 if not isinstance (num_threads , int ) or num_threads <= 0 :
308309 raise ValueError ("Number of threads must be integer and > 0" )
310+ if not isinstance (iteratively_update_sigma , bool ):
311+ raise TypeError ("Iteratively update sigma must be a boolean" )
309312 return func (
310313 self ,
311314 subset_data ,
@@ -316,6 +319,7 @@ def wrapper(
316319 normalize_target_data ,
317320 target_custom_scale_factor ,
318321 num_threads ,
322+ iteratively_update_sigma ,
319323 )
320324
321325 return wrapper
You can’t perform that action at this time.
0 commit comments