// Licensed to the .NET Foundation under one or more agreements.
// The .NET Foundation licenses this file to you under the MIT license.
// See the LICENSE file in the project root for more information.
using System;
using Microsoft.ML.TestFrameworkCommon.Attributes;
using Microsoft.ML.TestFrameworkCommon.Utility;
namespace Microsoft.ML.TestFramework.Attributes
{
/// <summary>
/// A fact for tests requiring LightGBM.
/// </summary>
public sealed class LightGBMFactAttribute : EnvironmentSpecificFactAttribute
{
public LightGBMFactAttribute() : base("LightGBM is 64-bit only")
{
}
/// <inheritdoc />
protected override bool IsEnvironmentSupported()
{
return Environment.Is64BitProcess && NativeLibrary.NativeLibraryExists("lib_lightgbm");
}
}
}
|