Minor fixes

NKI-AI · Sep 21, 2023 · 8f60d39 · 8f60d39
1 parent 6a4ea02
commit 8f60d39
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git 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
@@ -1,5 +1,6 @@
 # coding=utf-8
 # Copyright (c) DIRECT Contributors
+from __future__ import annotations
 
 from dataclasses import dataclass
 from typing import Optional