def load_feature_definitions_from_schema()

in pyspark-sdk/src/feature_store_pyspark/FeatureStoreManager.py [0:0]

• 6 lines of code
• 1 McCabe index (conditional complexity)

    def load_feature_definitions_from_schema(self, input_data_frame: DataFrame):
        """
        Load feature definitions according to the schema of input DataFrame.

        :param input_data_frame (DataFrame): input Spark DataFrame to be loaded.

        :return: list of feature definitions loaded from input DataFrame.
        """
        java_feature_definitions = self._call_java("loadFeatureDefinitionsFromSchema", input_data_frame)
        return list(map(lambda definition: {
            "FeatureName": definition.featureName(),
            "FeatureType": definition.featureType().toString()
        }, java_feature_definitions))