From 590d9d7f698796e70e05b013ba493605d1817d01 Mon Sep 17 00:00:00 2001
From: Joey Ballentine <joeyjballentine@gmail.com>
Date: Sat, 11 May 2024 12:16:51 -0400
Subject: [PATCH] Update tests

---
 tests/__snapshots__/test_PLKSR.ambr | 24 ++++++++++++++++++++++--
 1 file changed, 22 insertions(+), 2 deletions(-)

diff --git a/tests/__snapshots__/test_PLKSR.ambr b/tests/__snapshots__/test_PLKSR.ambr
index 3265ea21..06d988b3 100644
--- a/tests/__snapshots__/test_PLKSR.ambr
+++ b/tests/__snapshots__/test_PLKSR.ambr
@@ -13,6 +13,10 @@
     supports_bfloat16=True,
     supports_half=False,
     tags=list([
+      '64dim',
+      '28nb',
+      '17ks',
+      'DCCM',
     ]),
     tiling=<ModelTiling.SUPPORTED: 1>,
   )
@@ -31,6 +35,10 @@
     supports_bfloat16=True,
     supports_half=False,
     tags=list([
+      '64dim',
+      '28nb',
+      '17ks',
+      'DCCM',
     ]),
     tiling=<ModelTiling.SUPPORTED: 1>,
   )
@@ -49,6 +57,10 @@
     supports_bfloat16=True,
     supports_half=False,
     tags=list([
+      '64dim',
+      '28nb',
+      '17ks',
+      'DCCM',
     ]),
     tiling=<ModelTiling.SUPPORTED: 1>,
   )
@@ -57,7 +69,7 @@
   ImageModelDescriptor(
     architecture=PLKSRArch(
       id='PLKSR',
-      name='RealPLKSR',
+      name='PLKSR',
     ),
     input_channels=3,
     output_channels=3,
@@ -67,6 +79,10 @@
     supports_bfloat16=True,
     supports_half=False,
     tags=list([
+      '64dim',
+      '28nb',
+      '17ks',
+      'Real',
     ]),
     tiling=<ModelTiling.SUPPORTED: 1>,
   )
@@ -75,7 +91,7 @@
   ImageModelDescriptor(
     architecture=PLKSRArch(
       id='PLKSR',
-      name='RealPLKSR',
+      name='PLKSR',
     ),
     input_channels=3,
     output_channels=3,
@@ -85,6 +101,10 @@
     supports_bfloat16=True,
     supports_half=False,
     tags=list([
+      '64dim',
+      '28nb',
+      '17ks',
+      'Real',
     ]),
     tiling=<ModelTiling.SUPPORTED: 1>,
   )