assert str(model.device) == selected_device assert embeddings.shape == (2, model.get_embedding_dimension()) print(f"available devices: {', '.join(devices)}") print(f"selected device: {model.device}") print(f"embedding shape: {embeddings.shape}")