def dataset()

in nl2sql/datasets/standard.py [0:0]

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

    def dataset(self, **kwargs) -> Dataset:
        """
        Creates and returns a Dataset object based on the specified Spider split
        """
        return Dataset.from_connection_strings(
            name_connstr_map=dict(
                {
                    (i["db_id"], i["conn_str"])
                    for i in self.fetch_raw_data(
                        split=kwargs["split"], strict=kwargs.get("strict", False)
                    )
                }
            ),
            **kwargs,
        )