Skip to content

Commit 10596d8

Browse files
authored
Merge pull request #683 from cgre-aachen/fix_pd_warning
[WIP] Fix pd warning
2 parents b0c51a6 + ccf72e0 commit 10596d8

File tree

1 file changed

+16
-28
lines changed

1 file changed

+16
-28
lines changed

gempy/core/model.py

+16-28
Original file line numberDiff line numberDiff line change
@@ -519,11 +519,9 @@ def add_features(self, features_list: Union[str, list], reset_order_series=True)
519519
520520
"""
521521
self._stack.add_series(features_list, reset_order_series)
522-
self._surfaces.df['series'].cat.add_categories(features_list, inplace=True)
523-
self._surface_points.df['series'].cat.add_categories(features_list,
524-
inplace=True)
525-
self._orientations.df['series'].cat.add_categories(features_list,
526-
inplace=True)
522+
self._surfaces.df['series'] = self._surfaces.df['series'].cat.add_categories(features_list)
523+
self._surface_points.df['series'] = self._surface_points.df['series'].cat.add_categories(features_list)
524+
self._orientations.df['series'] = self._orientations.df['series'].cat.add_categories(features_list)
527525

528526
self.update_structure()
529527
self._interpolator.set_flow_control()
@@ -560,10 +558,9 @@ def delete_features(self, indices: Union[str, list], reset_order_features=True,
560558
self._surfaces.df.groupby('series').get_group(s)['surface'],
561559
remove_data=remove_data)
562560

563-
self._surfaces.df['series'].cat.remove_categories(indices, inplace=True)
564-
self._surface_points.df['series'].cat.remove_categories(indices,
565-
inplace=True)
566-
self._orientations.df['series'].cat.remove_categories(indices, inplace=True)
561+
self._surfaces.df['series'] = self._surfaces.df['series'].cat.remove_categories(indices)
562+
self._surface_points.df['series'] = self._surface_points.df['series'].cat.remove_categories(indices)
563+
self._orientations.df['series'] = self._orientations.df['series'].cat.remove_categories(indices)
567564
self.map_geometric_data_df(self._surface_points.df)
568565
self.map_geometric_data_df(self._orientations.df)
569566

@@ -601,12 +598,9 @@ def rename_features(self, new_categories: Union[dict, list]):
601598
602599
"""
603600
self._stack.rename_series(new_categories)
604-
self._surfaces.df['series'].cat.rename_categories(new_categories,
605-
inplace=True)
606-
self._surface_points.df['series'].cat.rename_categories(new_categories,
607-
inplace=True)
608-
self._orientations.df['series'].cat.rename_categories(new_categories,
609-
inplace=True)
601+
self._surfaces.df['series'] = self._surfaces.df['series'].cat.rename_categories(new_categories)
602+
self._surface_points.df['series'] = self._surface_points.df['series'].cat.rename_categories(new_categories)
603+
self._orientations.df['series'] = self._orientations.df['series'].cat.rename_categories(new_categories)
610604
return self._stack
611605

612606
@_setdoc(rename_features.__doc__, indent=False)
@@ -789,10 +783,8 @@ def set_surfaces_object(self):
789783
@_setdoc(Surfaces.add_surface.__doc__, indent=False)
790784
def add_surfaces(self, surface_list: Union[str, list], update_df=True):
791785
self._surfaces.add_surface(surface_list, update_df)
792-
self._surface_points.df['surface'].cat.add_categories(surface_list,
793-
inplace=True)
794-
self._orientations.df['surface'].cat.add_categories(surface_list,
795-
inplace=True)
786+
self._surface_points.df['surface'] = self._surface_points.df['surface'].cat.add_categories(surface_list)
787+
self._orientations.df['surface'] = self._orientations.df['surface'].cat.add_categories(surface_list)
796788
self.update_structure()
797789
return self._surfaces
798790

@@ -828,10 +820,8 @@ def delete_surfaces(self, indices: Union[str, Iterable[str]], update_id=True,
828820
self._orientations.df[
829821
self._orientations.df.surface.isin(surfaces_names)].index)
830822

831-
self._surface_points.df['surface'].cat.remove_categories(surfaces_names,
832-
inplace=True)
833-
self._orientations.df['surface'].cat.remove_categories(surfaces_names,
834-
inplace=True)
823+
self._surface_points.df['surface'] = self._surface_points.df['surface'].cat.remove_categories(surfaces_names)
824+
self._orientations.df['surface'] = self._orientations.df['surface'].cat.remove_categories(surfaces_names)
835825
self.map_geometric_data_df(self._surface_points.df)
836826
self.map_geometric_data_df(self._orientations.df)
837827
self._surfaces.colors.delete_colors(surfaces_names)
@@ -846,10 +836,8 @@ def delete_surfaces(self, indices: Union[str, Iterable[str]], update_id=True,
846836
def rename_surfaces(self, to_replace: Union[dict], **kwargs):
847837

848838
self._surfaces.rename_surfaces(to_replace, **kwargs)
849-
self._surface_points.df['surface'].cat.rename_categories(to_replace,
850-
inplace=True)
851-
self._orientations.df['surface'].cat.rename_categories(to_replace,
852-
inplace=True)
839+
self._surface_points.df['surface'] = self._surface_points.df['surface'].cat.rename_categories(to_replace)
840+
self._orientations.df['surface'] = self._orientations.df['surface'].cat.rename_categories(to_replace)
853841
return self._surfaces
854842

855843
@_setdoc(Surfaces.modify_order_surfaces.__doc__, indent=False)
@@ -934,7 +922,7 @@ def map_stack_to_surfaces(self,
934922

935923
# Here we remove the series that were not assigned to a surface
936924
if remove_unused_series is True:
937-
self._surfaces.df['series'].cat.remove_unused_categories(inplace=True)
925+
self._surfaces.df['series'] = self._surfaces.df['series'].cat.remove_unused_categories()
938926
unused_cat = self._stack.df.index[~self._stack.df.index.isin(
939927
self._surfaces.df['series'].cat.categories)]
940928
self._stack.delete_series(unused_cat)

0 commit comments

Comments
 (0)