responding to PR feedback

app/src/main/java/com/google/samples/apps/nowinandroid/ui/NiaApp.kt

@@ -41,7 +41,6 @@ import androidx.compose.material3.Scaffold
 import androidx.compose.material3.Text
 import androidx.compose.material3.windowsizeclass.ExperimentalMaterial3WindowSizeClassApi
 import androidx.compose.material3.windowsizeclass.WindowSizeClass
-import androidx.compose.material3.windowsizeclass.WindowSizeClass.Companion
 import androidx.compose.runtime.Composable
 import androidx.compose.ui.ExperimentalComposeUiApi
 import androidx.compose.ui.Modifier
@@ -127,8 +126,7 @@ fun NiaApp(
                         )
                     }
 
-                    if(!LocalInspectionMode.current)
-                    {
+                    if (!LocalInspectionMode.current) {
                         NiaNavHost(
                             navController = appState.navController,
                             onBackClick = appState::onBackClick,
@@ -137,9 +135,7 @@ fun NiaApp(
                                 .padding(padding)
                                 .consumedWindowInsets(padding)
                         )
-                    }
-                    else
-                    {
+                    } else {
                         BoxWithConstraints {
                             NiaTheme {
                                 ForYouScreen(
@@ -242,7 +238,7 @@ private fun NiaBottomBar(
 }
 @ReferenceDevicePreviews
 @Composable
-fun NiaAppLayoutPreview(){
+fun NiaAppLayoutPreview() {
     val widthDp = Dp(LocalConfiguration.current.screenWidthDp.toFloat())
     val heightDp = Dp(LocalConfiguration.current.screenHeightDp.toFloat())
     val windowSizeClass = WindowSizeClass.calculateFromSize(DpSize(widthDp, heightDp))

core/ui/src/main/java/com/google/samples/apps/nowinandroid/core/ui/ReferenceDevices.kt

@@ -16,10 +16,13 @@
 
 package com.google.samples.apps.nowinandroid.core.ui
 
-import androidx.compose.ui.tooling.preview.Preview
-
 import androidx.annotation.StringDef
+import androidx.compose.ui.tooling.preview.Preview
 
+/*
+ * TODO: Remove this file and instead use androidx.compose.ui.tooling.preview once this bug is
+ *  fixed - https://issuetracker.google.com/issues/254528382
+*/
 /**
  * List with the pre-defined devices available to be used in the preview.
  */