def load_dataset()

in gke/load-embeddings/main.py [0:0]

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

def load_dataset(location) -> pd.DataFrame:
    """Loads the dataset from the specified location"""
    df = pd.read_csv(location)
    df = df.loc[:, ["product_id", "product_name", "description", "list_price"]]
    df = df.dropna()
    return df