@@ -10,15 +10,13 @@
 #
 from __future__ import annotations
 
-import numpy as np
 import torch
 import torch.nn as nn
-import torch.nn.functional as F
 
 from deeplabcut.pose_estimation_pytorch.models.criterions import utils
 from deeplabcut.pose_estimation_pytorch.models.criterions.base import (
-    BaseCriterion,
     CRITERIONS,
+    BaseCriterion,
 )
Original file line number	Diff line number	Diff line change
`@@ -10,15 +10,13 @@`
`10`	`10`	`#`
`11`	`11`	`from __future__ import annotations`
`12`	`12`
`13`		`-import numpy as np`
`14`	`13`	`import torch`
`15`	`14`	`import torch.nn as nn`
`16`		`-import torch.nn.functional as F`
`17`	`15`
`18`	`16`	`from deeplabcut.pose_estimation_pytorch.models.criterions import utils`
`19`	`17`	`from deeplabcut.pose_estimation_pytorch.models.criterions.base import (`
`20`		`- BaseCriterion,`
`21`	`18`	`CRITERIONS,`
	`19`	`+ BaseCriterion,`
`22`	`20`	`)`
`23`	`21`
`24`	`22`