code: def setUpClass(cls):
      cls.evadb = get_evadb_for_testing()
      # reset the catalog manager before running each test
      cls.evadb.catalog().reset()
  
      create_table_query = """
             CREATE TABLE IF NOT EXISTS HomeRentals (
                 number_of_rooms INTEGER,
                 number_of_bathrooms INTEGER,
                 sqft INTEGER,
                 location TEXT(128),
                 days_on_market INTEGER,
                 initial_price INTEGER,
                 neighborhood TEXT(128),
                 rental_price FLOAT(64,64)
             );"""
      execute_query_fetch_all(cls.evadb, create_table_query)
  
      path = f"{EvaDB_ROOT_DIR}/data/ludwig/home_rentals.csv"
      load_query = f"LOAD CSV '{path}' INTO HomeRentals;"
      execute_query_fetch_all(cls.evadb, load_query)
  
      # Load data for classification tasks.
      create_table_query = """
             CREATE TABLE IF NOT EXISTS Employee (
                 education TEXT(128),
                 joining_year INTEGER,
                 city TEXT(128),
                 payment_tier INTEGER,
                 age INTEGER,
                 gender TEXT(128),
                 ever_benched TEXT(128),
                 experience_in_current_domain INTEGER,
                 leave_or_not INTEGER
             );"""
      execute_query_fetch_all(cls.evadb, create_table_query)
  
      path = f"{EvaDB_ROOT_DIR}/data/classification/Employee.csv"
      load_query = f"LOAD CSV '{path}' INTO Employee;"
      execute_query_fetch_all(cls.evadb, load_query)
  
file_path: evadb/test/integration_tests/long/test_model_train.py
imports: import unittest
  from test.markers import ludwig_skip_marker, sklearn_skip_marker, xgboost_skip_marker
  from test.util import get_evadb_for_testing, shutdown_ray
  import pytest
  from evadb.configuration.constants import EvaDB_ROOT_DIR
  from evadb.server.command_handler import execute_query_fetch_all
label_desc: Use image node_id 1 for calling the ModelTrainTests obj's underlying member method code with example usage: obj.setUpClass(cls) without return types