def convert_df_type_to_simple_type()

in src/smexperiments/tracker.py [0:0]

• 12 lines of code
• 3 McCabe index (conditional complexity)

    def convert_df_type_to_simple_type(cls, data_frame_type):
        """Converts a dataframe type to a type for rendering a table in Studio.

        Args:
            data_frame_type (str): The pandas type.

        Returns:
            str: The type of the table field.
        """

        type_pairs = [
            ("datetime", "datetime"),
            ("float", "number"),
            ("int", "number"),
            ("uint", "number"),
            ("boolean", "boolean"),
        ]
        for pair in type_pairs:
            if str(data_frame_type).lower().startswith(pair[0]):
                return pair[1]
        return "string"