deeplearning/projects/flashlight-sequence

Reviewed By: azad-meta

Differential Revision: D53935505

fbshipit-source-id: 65996e1616d06d25cc03394dc26134154da63cab

Author: generatedunixname89002005287564

bindings/python/flashlight/lib/sequence/criterion.py

@@ -6,33 +6,13 @@
 LICENSE file in the root directory of this source tree.
 """
 
-from .flashlight_lib_sequence_criterion import (
-    CpuForceAlignmentCriterion,
-    CpuFullConnectionCriterion,
-    CpuViterbiPath,
-    CriterionScaleMode,
-)
-
 
 have_torch = False
 try:
-    import torch
 
     have_torch = True
 except ImportError:
     pass
 
 if have_torch:
-    from flashlight.lib.sequence.criterion_torch import (
-        ASGLoss,
-        check_tensor,
-        create_workspace,
-        FACFunction,
-        FCCFunction,
-        get_cuda_stream_as_bytes,
-        get_data_ptr_as_bytes,
-        run_backward,
-        run_direction,
-        run_forward,
-        run_get_workspace_size,
-    )
+    pass

bindings/python/test/test_import.py

@@ -11,13 +11,6 @@
 
 class ImportTestCase(unittest.TestCase):
     def test_import_lib_sequence(self) -> None:
-        from flashlight.lib.sequence import criterion
-        from flashlight.lib.sequence.criterion import (
-            CpuForceAlignmentCriterion,
-            CpuFullConnectionCriterion,
-            CpuViterbiPath,
-            CriterionScaleMode,
-        )
 
         if os.getenv("USE_CUDA", "OFF").upper() not in [
             "OFF",
@@ -26,11 +19,7 @@ def test_import_lib_sequence(self) -> None:
             "FALSE",
             "N",
         ]:
-            from flashlight.lib.sequence.flashlight_lib_sequence_criterion import (
-                CudaForceAlignmentCriterion,
-                CudaFullConnectionCriterion,
-                CudaViterbiPath,
-            )
+            pass
         else:
             logging.info("Flashlight Sequence bindings built without CUDA")