From ceff0823405abf397471a5f95a920a97a6e8e695 Mon Sep 17 00:00:00 2001 From: SCM Date: Mon, 26 Jun 2023 17:46:21 +0100 Subject: [PATCH] os.sep rather than '\' or analysis_example.py --- examples/analysis_example.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/examples/analysis_example.py b/examples/analysis_example.py index f716c08..12d029f 100644 --- a/examples/analysis_example.py +++ b/examples/analysis_example.py @@ -1,5 +1,5 @@ import torch - +import os from molearn.models.foldingnet import AutoEncoder from molearn.analysis import MolearnAnalysis from molearn.data import PDBData @@ -7,7 +7,7 @@ print("> Loading network parameters...") -fname = 'xbb_foldingnet_checkpoints\\checkpoint_no_optimizer_state_dict_epoch167_loss0.003259085263643.ckpt' +fname = f'xbb_foldingnet_checkpoints{os.sep}checkpoint_no_optimizer_state_dict_epoch167_loss0.003259085263643.ckpt' checkpoint = torch.load(fname, map_location=torch.device('cpu')) net = AutoEncoder(**checkpoint['network_kwargs']) net.load_state_dict(checkpoint['model_state_dict']) @@ -29,8 +29,8 @@ # by defining the manual see and loading the dataset in the same order as when #the neural network was trained, the same train-test split will be obtained data = PDBData() -data.import_pdb('data\\MurD_closed_selection.pdb') -data.import_pdb('data\\MurD_open_selection.pdb') +data.import_pdb(f'data{os.sep}MurD_closed_selection.pdb') +data.import_pdb(f'data{os.sep}MurD_open_selection.pdb') data.fix_terminal() data.atomselect(atoms = ['CA', 'C', 'N', 'CB', 'O']) data.prepare_dataset()