Flatsome Change Product Columns per Category

If you want to change the product columns per category or on your shop page in Flatsome, use this snippet:

add_filter('theme_mod_category_row_count', function($columns) {
	if(is_shop()) {
		return 4;
	} elseif(is_product_category('YOUR_CATEGORY_SLUG')) {
		return 4;
	}
	return $columns;
}, 10, 1);