diff --git a/pysipfenn/core/pysipfenn.py b/pysipfenn/core/pysipfenn.py index daa63ec..a7c0ad2 100644 --- a/pysipfenn/core/pysipfenn.py +++ b/pysipfenn/core/pysipfenn.py @@ -223,6 +223,7 @@ def downloadModels(self, network: str = 'all') -> None: # Fetch all if network == 'all': print('Fetching all networks!') + downloadableNets = [net for net in self.network_list if 'URL_ONNX' in self.models[net]] for net in self.network_list: if net not in self.network_list_available: if 'URL_ONNX' in self.models[net]: @@ -236,8 +237,8 @@ def downloadModels(self, network: str = 'all') -> None: print(f'{net} not detected on disk and ONNX URL has not been provided.') else: print(f'{net} detected on disk. Ready to use.') - if self.network_list == self.network_list_available: - print('All networks available!') + if downloadableNets == self.network_list_available: + print('All downloadable networks are now available!') else: print('Problem occurred.')