Import from the newly cleaned-up prototorch namespace

2021-06-14 20:08:08 +02:00
parent c87ed5ba8b
commit 69e5ff3243
10 changed files with 80 additions and 37 deletions
--- a/prototorch/models/callbacks.py
+++ b/prototorch/models/callbacks.py
@@ -4,8 +4,8 @@ import logging

 import pytorch_lightning as pl
 import torch
-from prototorch.components import Components

+from ..core.components import Components
 from .extras import ConnectionTopology