import torch
import mindsdb
from tests import basic_test
if __name__ == "__main__":
use_gpu_settings = [False]
if torch.cuda.is_available():
use_gpu_settings.append(True)
# Cycle through a few options:
for backend in ['lightwood']:
for use_gpu in use_gpu_settings:
print(f'use_gpu is set to {use_gpu}, backend is set to {backend}')
basic_test(backend=backend, use_gpu=use_gpu, IS_CI_TEST=True)
print(
'\n\n=============[Success]==============\n Finished running full test suite !\n=============[Success]==============\n\n'
)
from tests import basic_test
# Test with a few basic options
basic_test(backend='lightwood',use_gpu=False,ignore_columns=[])
print('\n\n=============[Success]==============\n Finished running quick test !\n=============[Success]==============\n\n')
from tests import basic_test
# Cycle through a few options:
for backend in ['lightwood', 'ludwig']:
for use_gpu in [False, True]:
print(f'use_gpu is set to {use_gpu}, backend is set to {backend}')
basic_test(backend=backend, use_gpu=use_gpu, ignore_columns=[])
# Try ignoring some columns
basic_test(backend='lightwood',
use_gpu=True,
ignore_columns=['days_on_market', 'number_of_bathrooms'])
print(
'\n\n=============[Success]==============\n Finished running full test suite !\n=============[Success]==============\n\n'
)
import torch
import mindsdb
from tests import basic_test
if __name__ == "__main__":
use_gpu_settings = []
if torch.cuda.is_available():
use_gpu_settings.append(True)
use_gpu_settings.append(False)
# Try ignoring some columns and running only the stats generator
run_extra = True
# Cycle through a few options:
for backend in ['lightwood']: #,'ludwig'
for use_gpu in use_gpu_settings:
print(f'use_gpu is set to {use_gpu}, backend is set to {backend}')
if run_extra:
basic_test(backend=backend,use_gpu=use_gpu,ignore_columns=['days_on_market','number_of_bathrooms'], IS_CI_TEST=True, run_extra=True)
run_extra = False
else:
basic_test(backend=backend,use_gpu=use_gpu,ignore_columns=[], IS_CI_TEST=True)
print('\n\n=============[Success]==============\n Finished running full test suite !\n=============[Success]==============\n\n')
import torch
import mindsdb
from tests import basic_test
if __name__ == "__main__":
use_gpu_settings = []
if torch.cuda.is_available():
use_gpu_settings.append(True)
use_gpu_settings.append(False)
# Cycle through a few options:
for backend in ['lightwood','ludwig']:
for use_gpu in use_gpu_settings:
print(f'use_gpu is set to {use_gpu}, backend is set to {backend}')
basic_test(backend=backend,use_gpu=use_gpu,ignore_columns=[], IS_CI_TEST=True)
# Try ignoring some columns and running only the stats generator
basic_test(backend='lightwood',use_gpu=use_gpu_settings[0],ignore_columns=['days_on_market','number_of_bathrooms'],run_extra=True, IS_CI_TEST=True)
print('\n\n=============[Success]==============\n Finished running full test suite !\n=============[Success]==============\n\n')
from tests import basic_test
import torch
# Test with a few basic options
if __name__ == "__main__":
basic_test(backend='lightwood',use_gpu=torch.cuda.is_available(),ignore_columns=[], IS_CI_TEST=True)
print('\n\n=============[Success]==============\n Finished running quick test !\n=============[Success]==============\n\n')
