[api] Remove unnecessary DeviceType interface

Change-Id: Ife754d8b82371c1cd86d8201d003fe8f7995ed92
deepjavalibrary · Aug 31, 2022 · 22a0d12 · 22a0d12
1 parent 319d80c
commit 22a0d12
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 42 deletions.
diff --git a/api/src/main/java/ai/djl/DeviceType.java b/api/src/main/java/ai/djl/DeviceType.java
diff --git a/engines/mxnet/mxnet-engine/src/main/java/ai/djl/mxnet/engine/MxDeviceType.java b/engines/mxnet/mxnet-engine/src/main/java/ai/djl/mxnet/engine/MxDeviceType.java
@@ -13,10 +13,9 @@
 package ai.djl.mxnet.engine;
 
 import ai.djl.Device;
-import ai.djl.DeviceType;
 
 /** DeviceType is the MXNet equivalent of the types in {@link Device}. */
-public final class MxDeviceType implements DeviceType {
+public final class MxDeviceType {
 
  private static final String CPU_PINNED = "cpu_pinned";
 
@@ -38,7 +37,7 @@ public static int toDeviceType(Device device) {
  } else if (CPU_PINNED.equals(deviceType)) {
  return 3;
  } else {
- throw new IllegalArgumentException("Unsupported device: " + device.toString());
+ throw new IllegalArgumentException("Unsupported device: " + device);
  }
  }
 

diff --git a/engines/pytorch/pytorch-engine/src/main/java/ai/djl/pytorch/engine/PtDeviceType.java b/engines/pytorch/pytorch-engine/src/main/java/ai/djl/pytorch/engine/PtDeviceType.java
@@ -13,10 +13,9 @@
 package ai.djl.pytorch.engine;
 
 import ai.djl.Device;
-import ai.djl.DeviceType;
 
 /** DeviceType is the PyTorch equivalent of the types in {@link Device}. */
-public final class PtDeviceType implements DeviceType {
+public final class PtDeviceType {
 
  private PtDeviceType() {}