diff --git a/direct/nn/lpd/lpd.py b/direct/nn/lpd/lpd.py index d1884c7d..04683b49 100644 --- a/direct/nn/lpd/lpd.py +++ b/direct/nn/lpd/lpd.py @@ -179,7 +179,7 @@ def __init__( ) else: raise NotImplementedError( - f"XPDNet is currently implemented only with primal_model_architecture == 'MWCNN', 'UNET' or 'NORMUNET. " + f"XPDNet is currently implemented only with primal_model_architecture == 'MWCNN', 'UNET' or 'NORMUNET." f"Got {primal_model_architecture}." ) dual_model: nn.Module diff --git a/direct/nn/vsharp/config.py b/direct/nn/vsharp/config.py index 622e9bba..64d45f31 100644 --- a/direct/nn/vsharp/config.py +++ b/direct/nn/vsharp/config.py @@ -1,5 +1,6 @@ # coding=utf-8 # Copyright (c) DIRECT Contributors +from __future__ import annotations from dataclasses import dataclass from typing import Optional