Fix test import of metrax.logging. (#121)

* fix

* also fix tensorboard

Author: jshin1394

src/metrax/logging/tensorboard_backend_test.py

@@ -15,7 +15,7 @@
 from unittest import mock
 
 from absl.testing import absltest
-from metrax import logging as metrax_logging
+import metrax.logging as metrax_logging
 
 TensorboardBackend = metrax_logging.TensorboardBackend
 

src/metrax/logging/wandb_backend_test.py

@@ -4,7 +4,7 @@
 from unittest import mock
 
 from absl.testing import absltest
-from metrax import logging as metrax_logging
+import metrax.logging as metrax_logging
 
 WandbBackend = metrax_logging.WandbBackend