fix tensorflow_model_optimization.bzl

Added missing definition, and fixed py_strict_test rule.

PiperOrigin-RevId: 370028146

Author: teijeong

tensorflow_model_optimization/tensorflow_model_optimization.bzl

@@ -30,6 +30,10 @@ def pytype_strict_binary(name, **kwargs):
 def pytype_strict_library(name, **kwargs):
     native.py_library(name = name, **kwargs)
 
+# Placeholder to use until bazel supports pytype_library.
+def pytype_library(name, **kwargs):
+    native.py_library(name = name, **kwargs)
+
 # Placeholder to use until bazel supports py_strict_test.
 def py_strict_test(name, **kwargs):
-    py_test(name = name, **kwargs)
+    native.py_test(name = name, **kwargs)