@@ -466,19 +466,15 @@ def accessor_entry_point_loader() -> None:
466
466
df.myplugin.do_something() # Calls MyPluginAccessor.do_something()
467
467
"""
468
468
469
- PD_OBJECTS_ENTRYPOINTS : list [str ] = [
470
- "pandas.DataFrame.accessor" ,
471
- "pandas.Series.accessor" ,
472
- "pandas.Index.accessor" ,
473
- ]
474
-
475
469
ACCESSOR_REGISTRY_FUNCTIONS : dict [str , Callable ] = {
476
470
"pandas.DataFrame.accessor" : register_dataframe_accessor ,
477
471
"pandas.Series.accessor" : register_series_accessor ,
478
472
"pandas.Index.accessor" : register_index_accessor ,
479
473
}
480
474
481
- for pd_obj_entrypoint in PD_OBJECTS_ENTRYPOINTS :
475
+ pd_objects_entrypoints : list [str ] = ACCESSOR_REGISTRY_FUNCTIONS .keys ()
476
+
477
+ for pd_obj_entrypoint in pd_objects_entrypoints :
482
478
accessors : EntryPoints = entry_points (group = pd_obj_entrypoint )
483
479
accessor_package_dict : dict [str , str ] = {}
484
480
@@ -488,10 +484,9 @@ def accessor_entry_point_loader() -> None:
488
484
489
485
# Verifies duplicated accessor names
490
486
if new_accessor .name in accessor_package_dict :
491
- loaded_pkg_name : str = accessor_package_dict .get (new_accessor .name )
492
-
493
- if loaded_pkg_name is None :
494
- loaded_pkg_name : str = "Unknown"
487
+ loaded_pkg_name : str = accessor_package_dict .get (
488
+ new_accessor .name , "Unknown"
489
+ )
495
490
496
491
warnings .warn (
497
492
"Warning: you have two accessors with the same name:"
0 commit comments