Coverage for tests / unit / nbutils / test_configure_notebook_torch.py: 100%

1import torch 

2 

3from muutils.nbutils.configure_notebook import configure_notebook 

4 

5 

6def test_configure_notebook(): 

7 """Test configure_notebook which returns a torch.device.""" 

8 device = configure_notebook(seed=42, plot_mode="ignore") 

9 assert isinstance(device, torch.device)