import random, polars as pl
from great_tables import GT

random.seed(42)
col_1 =  [random.uniform(-1.0, 1.0) for a in list(range(7))]
col_2 =  [random.uniform(-1.0, 1.0) for a in list(range(7))]
df = pl.DataFrame({'COL_1': col_1,'COL_2': col_2})

print(df.head(7))
my_gt = (
    GT(df)
    .tab_header(title = 'Positive, Negative Cosine') 
    # Test with.fmt_number invoked, and with .fmt_number commented out
    # .fmt_number(columns=['COL_1', 'COL_2'], decimals=3)
)

# .save fails when great_table .fmt_number was used
my_gt.save('Random.png', window_size=(6, 6))