fix: sphinx class reference

Author: [email protected]

src/sagemaker/feature_group_utils.py

@@ -30,7 +30,7 @@
 
 
 def _get_session_from_role(role: str, region: str) -> Session:
-    """Method use to get the :class:~`sagemaker.session.Session`  from a role and a region.
+    """Method use to get the ":class:~`sagemaker.session.Session`"  from a role and a region.
 
     Helpful in case it's invoke from a session with a role without permission it can assume
     another role temporarily to perform certain taks.
@@ -89,8 +89,7 @@ def get_feature_group_as_dataframe(
     verbose: bool = True,
     **pandas_read_csv_kwargs,
 ) -> DataFrame:
-    """Get a :class:~`sagemaker.feature_store.feature_group.FeatureGroup`
-    as a pandas.DataFrame
+    """Get a ":class:~`sagemaker.feature_store.feature_group.FeatureGroup`" as a pandas.DataFrame
 
     Description:
         Method to run an athena query over a Feature Group in a Feature Store
@@ -108,7 +107,7 @@ def get_feature_group_as_dataframe(
                     'SELECT * FROM "sagemaker_featurestore"."#{table}"'
         athena_bucket (str): Amazon S3 bucket for running the query
         role (str): role of the account used to extract data from feature store
-        session (str): :class:~`sagemaker.session.Session`
+        session (str): ":class:"~`sagemaker.session.Session`"
                         of SageMaker used to work with the feature store
         event_time_feature_name (str): eventTimeId feature. Mandatory only if the
                                         latest ingestion is True
@@ -175,8 +174,7 @@ def get_feature_group_as_dataframe(
 
 
 def _format_column_names(data: pandas.DataFrame) -> pandas.DataFrame:
-    """Formats the column names in a valid way for
-    :class:~`sagemaker.feature_store.feature_group.FeatureGroup`
+    """Formats the column names in a valid way for ":class:~`sagemaker.feature_store.feature_group.FeatureGroup`"
 
     Module to format correctly the name of the columns of a DataFrame
     to later generate the features names of a Feature Group
@@ -220,7 +218,7 @@ def prepare_fg_from_dataframe_or_file(
     **pandas_read_csv_kwargs,
 ) -> FeatureGroup:
     """Module to prepare a dataframe before creating a
-    :class:~`sagemaker.feature_store.feature_group.FeatureGroup`
+        ":class:"~`sagemaker.feature_store.feature_group.FeatureGroup`"
 
     Function to prepare a dataframe for creating a Feature Group from a pandas.DataFrame
     or a path to a file with proper dtypes, feature names and mandatory features (record_id,
@@ -244,8 +242,8 @@ def prepare_fg_from_dataframe_or_file(
         session (str): session of SageMaker used to work with the feature store
 
     Returns:
-        :class:~`sagemaker.feature_store.feature_group.FeatureGroup`:
-        FG prepared with all the methods and definitions properly defined
+        ":class:"~`sagemaker.feature_store.feature_group.FeatureGroup`": FG prepared with all
+            the methods and definitions properly defined
     """
 
     logger.setLevel(logging.WARNING)