import torch
import pickle

# Example SMILES strings
smiles_list = ["CC(=O)OC1=CC=CC=C1C(=O)O", "C1CCCCC1", "CCN(CC)CC"]

# Create 256-D random embeddings
embeddings_dict = {smi: torch.randn(256) for smi in smiles_list}

# Save to embeddings.pkl
with open("data/processed/embeddings.pkl", "wb") as f:
    pickle.dump(embeddings_dict, f)

print("Saved dummy embeddings to data/processed/embeddings.pkl")