Update truth usage to compile with truth-1.0.1 APIs.

Bug: 168765701
Test: m checkbuild
Merged-In: Ie8e7a33322f711c2a2e6a2a7274edfa216d54f87
Change-Id: Ie8e7a33322f711c2a2e6a2a7274edfa216d54f87
(cherry picked from commit 21217cdf435f5961605c1d2806a80c9e110b7c8c)
diff --git a/java/tests/instrumentation/src/com/android/textclassifier/subjects/EntitySubject.java b/java/tests/instrumentation/src/com/android/textclassifier/subjects/EntitySubject.java
index 30a9dca..569143f 100644
--- a/java/tests/instrumentation/src/com/android/textclassifier/subjects/EntitySubject.java
+++ b/java/tests/instrumentation/src/com/android/textclassifier/subjects/EntitySubject.java
@@ -24,7 +24,7 @@
 import javax.annotation.Nullable;
 
 /** Test helper for checking {@link com.android.textclassifier.Entity} results. */
-public final class EntitySubject extends Subject<EntitySubject, Entity> {
+public final class EntitySubject extends Subject {
 
   private static final float TOLERANCE = 0.0001f;